body_25 neck index error

Author: davidpagnon

Sports2D/Sports2D.ipynb

@@ -553,7 +553,7 @@
         "video_names = [os.path.basename(video_name) for video_name in video_names]\n",
         "pose_model = 'BODY_25'\n",
         "config_dict.get('project').update({'video_dir':video_path})\n",
-        "config_dict.get('project').update({'video_file':video_names})\n",
+        "config_dict.get('project').update({'video_files':video_names})\n",
         "config_dict.get('project').update({'result_dir':result_path})\n",
         "config_dict.get('pose').update({'pose_algo':'OPENPOSE'})\n",
         "config_dict.get('pose').get('OPENPOSE').update({'openpose_model':pose_model})\n",
@@ -972,9 +972,9 @@
         "                      </video>'''.format(width, height, video_encoded.decode('ascii')))\n",
         "\n",
         "if show_last == 'Yes':\n",
-        "  vid_to_show = os.path.join(result_path, os.path.splitext(video_name)[0] + '_' + pose_model + os.path.splitext(video_name)[1])\n",
+        "  vid_to_show = os.path.join(result_path, os.path.splitext(video_names[-1])[0] + '_' + pose_model + os.path.splitext(video_names[-1])[1])\n",
         "  if not os.path.isfile(vid_to_show):\n",
-        "    vid_to_show = os.path.join(result_path, os.path.splitext(video_name)[0] + '_' + pose_model + '.mp4')\n",
+        "    vid_to_show = os.path.join(result_path, os.path.splitext(video_names[-1])[0] + '_' + pose_model + '.mp4')\n",
         "else:\n",
         "  vid_to_show = os.path.join(result_path, show_video_name)\n",
         "  if not os.path.isfile(vid_to_show):\n",
@@ -2746,7 +2746,7 @@
         "import plotly.express as px\n",
         "\n",
         "#@markdown Display knee angle results of the first person detected in the last uploaded video:\n",
-        "plot_last = 'No' #@param [\"Yes\", \"No\"]\n",
+        "plot_last = 'Yes' #@param [\"Yes\", \"No\"]\n",
         "\n",
         "#@markdown </br>\n",
         "\n",
@@ -2762,13 +2762,13 @@
         "\n",
         "\n",
         "if plot_last == 'Yes':\n",
-        "  csv_to_show = os.path.join(result_path, os.path.splitext(video_name)[0] + '_' + pose_model + '_angles.csv')\n",
+        "  csv_to_show = os.path.join(result_path, os.path.splitext(video_names[-1])[0] + '_' + pose_model + '_angles.csv')\n",
         "  if not os.path.isfile(csv_to_show):\n",
-        "    csv_to_show = os.path.join(result_path, os.path.splitext(video_name)[0] + '_' + pose_model + '_person0_angles.csv')\n",
+        "    csv_to_show = os.path.join(result_path, os.path.splitext(video_names[-1])[0] + '_' + pose_model + '_person0_angles.csv')\n",
         "  if not os.path.isfile(csv_to_show):\n",
-        "    csv_to_show = os.path.join(result_path, os.path.splitext(video_name)[0] + '_converted' + pose_model + '_angles.csv')\n",
+        "    csv_to_show = os.path.join(result_path, os.path.splitext(video_names[-1])[0] + '_converted' + pose_model + '_angles.csv')\n",
         "  if not os.path.isfile(csv_to_show):\n",
-        "    csv_to_show = os.path.join(result_path, os.path.splitext(video_name)[0] + '_converted' + pose_model + '_person0_angles.csv')\n",
+        "    csv_to_show = os.path.join(result_path, os.path.splitext(video_names[-1])[0] + '_converted' + pose_model + '_person0_angles.csv')\n",
         "else:\n",
         "  csv_to_show = os.path.join(result_path, plot_csv_name)\n",
         "  if not os.path.isfile(csv_to_show):\n",
@@ -2783,7 +2783,7 @@
         "  table_select = table[plot_variable_names]\n",
         "  table_select.columns = [' '.join(col).strip() for col in table_select.columns.values]\n",
         "  table_select = table_select.set_index(list(table_select)[0])\n",
-        "  fig = px.line(data_frame=table_select, width=1310, height=699)\n",
+        "  fig = px.line(data_frame=table_select, width=1310, height=699, title=os.path.basename(csv_to_show))\n",
         "  fig.show()\n",
         "except:\n",
         "  print('Variables could not be found in csv file')\n",

Sports2D/Utilities/skeletons.py

@@ -126,7 +126,7 @@
             ]),
         ]),
     ]),
-    Node("Neck", id=17, children=[
+    Node("Neck", id=1, children=[
         Node("Nose", id=0, children=[
                 Node("REye", id=15, children=[
                     Node("REar", id=17)