sports2d 0.8.10__py3-none-any.whl → 0.8.12__py3-none-any.whl

Sports2D/process.py

@@ -60,6 +60,7 @@ import copy
 import shutil
 import os
 import re
+import platform
 from importlib.metadata import version
 from datetime import datetime
 import itertools as it
@@ -82,6 +83,7 @@ from Sports2D.Utilities import filter
 from Sports2D.Utilities.common import *
 from Pose2Sim.common import *
 from Pose2Sim.skeletons import *
+from Pose2Sim.triangulation import indices_of_first_last_non_nan_chunks
 
 DEFAULT_MASS = 70
 DEFAULT_HEIGHT = 1.7
@@ -494,24 +496,25 @@ def draw_angles(img, valid_X, valid_Y, valid_angles, valid_X_flipped, keypoints_
                             right_angle = True if ang_params[2]==90 else False
                             
                             # Draw angle
-                            if len(ang_coords) == 2: # segment angle
-                                app_point, vec = draw_segment_angle(img, ang_coords, flip)
-                            else: # joint angle
-                                app_point, vec1, vec2 = draw_joint_angle(img, ang_coords, flip, right_angle)
-        
-                            # Write angle on body
-                            if 'body' in display_angle_values_on:
+                            if not np.any(np.isnan(ang_coords)):
                                 if len(ang_coords) == 2: # segment angle
-                                    write_angle_on_body(img, ang, app_point, vec, np.array([1,0]), dist=20, color=(255,255,255), fontSize=fontSize, thickness=thickness)
+                                    app_point, vec = draw_segment_angle(img, ang_coords, flip)
                                 else: # joint angle
-                                    write_angle_on_body(img, ang, app_point, vec1, vec2, dist=40, color=(0,255,0), fontSize=fontSize, thickness=thickness)
-
-                            # Write angle as a list on image with progress bar
-                            if 'list' in display_angle_values_on:
-                                if len(ang_coords) == 2: # segment angle
-                                    ang_label_line = write_angle_as_list(img, ang, ang_name, person_label_position, ang_label_line, color = (255,255,255), fontSize=fontSize, thickness=thickness)
-                                else:
-                                    ang_label_line = write_angle_as_list(img, ang, ang_name, person_label_position, ang_label_line, color = (0,255,0), fontSize=fontSize, thickness=thickness)
+                                    app_point, vec1, vec2 = draw_joint_angle(img, ang_coords, flip, right_angle)
+        
+                                # Write angle on body
+                                if 'body' in display_angle_values_on:
+                                    if len(ang_coords) == 2: # segment angle
+                                        write_angle_on_body(img, ang, app_point, vec, np.array([1,0]), dist=20, color=(255,255,255), fontSize=fontSize, thickness=thickness)
+                                    else: # joint angle
+                                        write_angle_on_body(img, ang, app_point, vec1, vec2, dist=40, color=(0,255,0), fontSize=fontSize, thickness=thickness)
+
+                                # Write angle as a list on image with progress bar
+                                if 'list' in display_angle_values_on:
+                                    if len(ang_coords) == 2: # segment angle
+                                        ang_label_line = write_angle_as_list(img, ang, ang_name, person_label_position, ang_label_line, color = (255,255,255), fontSize=fontSize, thickness=thickness)
+                                    else:
+                                        ang_label_line = write_angle_as_list(img, ang, ang_name, person_label_position, ang_label_line, color = (0,255,0), fontSize=fontSize, thickness=thickness)
 
     return img
 
@@ -788,8 +791,11 @@ def pose_plots(trc_data_unfiltered, trc_data, person_id):
     OUTPUT:
     - matplotlib window with tabbed figures for each keypoint
     '''
-    
-    mpl.use('qt5agg')
+
+    os_name = platform.system()
+    if os_name == 'Windows':
+        mpl.use('qt5agg') # windows
+
     mpl.rc('figure', max_open_warning=0)
 
     keypoints_names = trc_data.columns[1::3]
@@ -799,7 +805,10 @@ def pose_plots(trc_data_unfiltered, trc_data, person_id):
 
     for id, keypoint in enumerate(keypoints_names):
         f = plt.figure()
-        f.canvas.manager.window.setWindowTitle(keypoint + ' Plot')
+        if os_name == 'Windows':
+            f.canvas.manager.window.setWindowTitle(keypoint + ' Plot') # windows
+        elif os_name == 'Darwin':  # macOS
+            f.canvas.manager.set_window_title(keypoint + ' Plot') # mac
 
         axX = plt.subplot(211)
         plt.plot(trc_data_unfiltered.iloc[:,0], trc_data_unfiltered.iloc[:,id*3+1], label='unfiltered')
@@ -832,16 +841,22 @@ def angle_plots(angle_data_unfiltered, angle_data, person_id):
     - matplotlib window with tabbed figures for each angle
     '''
 
-    mpl.use('qt5agg')
+    os_name = platform.system()
+    if os_name == 'Windows':
+        mpl.use('qt5agg') # windows
     mpl.rc('figure', max_open_warning=0)
 
     angles_names = angle_data.columns[1:]
-
+    
     pw = plotWindow()
     pw.MainWindow.setWindowTitle('Person'+ str(person_id) + ' angles') # Main title
 
     for id, angle in enumerate(angles_names):
         f = plt.figure()
+        if os_name == 'Windows':
+            f.canvas.manager.window.setWindowTitle(angle + ' Plot') # windows
+        elif os_name == 'Darwin':  # macOS
+            f.canvas.manager.set_window_title(angle + ' Plot') # mac
         
         ax = plt.subplot(111)
         plt.plot(angle_data_unfiltered.iloc[:,0], angle_data_unfiltered.iloc[:,id+1], label='unfiltered')
@@ -1811,9 +1826,15 @@ def process_fun(config_dict, video_file, time_range, frame_rate, result_dir):
                     logging.info(f'- Person {i}: Interpolating missing sequences if they are smaller than {interp_gap_smaller_than} frames. Large gaps filled with {fill_large_gaps_with}.')
                     all_frames_X_person_interp = all_frames_X_person.apply(interpolate_zeros_nans, axis=0, args = [interp_gap_smaller_than, 'linear'])
                     all_frames_Y_person_interp = all_frames_Y_person.apply(interpolate_zeros_nans, axis=0, args = [interp_gap_smaller_than, 'linear'])
+
                     if fill_large_gaps_with.lower() == 'last_value':
-                        all_frames_X_person_interp = all_frames_X_person_interp.ffill(axis=0).bfill(axis=0)
-                        all_frames_Y_person_interp = all_frames_Y_person_interp.ffill(axis=0).bfill(axis=0)
+                        for col in all_frames_X_person_interp.columns:
+                            first_run_start, last_run_end = indices_of_first_last_non_nan_chunks(all_frames_Y_person_interp[col])
+                            for coord_df in [all_frames_X_person_interp, all_frames_Y_person_interp]:
+                                coord_df.loc[:first_run_start, col] = np.nan
+                                coord_df.loc[last_run_end:, col] = np.nan
+                                coord_df.loc[first_run_start:last_run_end, col] = coord_df.loc[first_run_start:last_run_end, col].ffill().bfill()
+
                     elif fill_large_gaps_with.lower() == 'zeros':
                         all_frames_X_person_interp.replace(np.nan, 0, inplace=True)
                         all_frames_Y_person_interp.replace(np.nan, 0, inplace=True)

{sports2d-0.8.10.dist-info → sports2d-0.8.12.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sports2d
-Version: 0.8.10
+Version: 0.8.12
 Summary: Compute 2D human pose and angles from a video or a webcam.
 Author-email: David Pagnon <contact@david-pagnon.com>
 Maintainer-email: David Pagnon <contact@david-pagnon.com>

{sports2d-0.8.10.dist-info → sports2d-0.8.12.dist-info}/RECORD

@@ -11,16 +11,16 @@ Content/sports2d_opensim.gif,sha256=XP1AcjqhbGcJknXUoNJjPWAwaM9ahZafbDgLWvzKJs4,
 Sports2D/Sports2D.ipynb,sha256=VnOVjIl6ndnCJTT13L4W5qTw4T-TQDF3jt3-wxnXDqM,2427047
 Sports2D/Sports2D.py,sha256=tbQi_d6GXMqFkSd1_FRxyl6oA4CXEJH0g-lEskmR4mI,33521
 Sports2D/__init__.py,sha256=BuUkPEdItxlkeqz4dmoiPwZLkgAfABJK3KWQ1ujTGwE,153
-Sports2D/process.py,sha256=UovoTjXSxlrTNnuog9MlQu_2HCAypovDVUKIUIZ8k_c,111353
+Sports2D/process.py,sha256=7qmRF8xi41A-_O5aeuvM1WTJLXCLzwWUvT5klku3qJE,112492
 Sports2D/Demo/Config_demo.toml,sha256=jiPGr6JbePu1tItgRxdAu45mwFqr3RcIi_tfhO2ik1g,14027
 Sports2D/Demo/demo.mp4,sha256=2aZkFxhWR7ESMEtXCT8MGA83p2jmoU2sp1ylQfO3gDk,3968304
 Sports2D/Utilities/__init__.py,sha256=BuUkPEdItxlkeqz4dmoiPwZLkgAfABJK3KWQ1ujTGwE,153
 Sports2D/Utilities/common.py,sha256=idMRmesFv5BPX-5g3z5dOVa7SpS_8tNgijvGrOZlR-k,11185
 Sports2D/Utilities/filter.py,sha256=rfZcqofjllKI_5ovZTKEAmyjOZpB_PzbAJ0P874T8Ak,4973
 Sports2D/Utilities/tests.py,sha256=3-slY2teKd78q0NzfJ2H84xRFSNXvGyDwew27oHJfrY,4740
-sports2d-0.8.10.dist-info/licenses/LICENSE,sha256=f4qe3nE0Y7ltJho5w-xAR0jI5PUox5Xl-MsYiY7ZRM8,1521
-sports2d-0.8.10.dist-info/METADATA,sha256=5qmY0gVTZV3vZkr5n9QizGfNwbIHDT5x2tqUoSdvvMs,38443
-sports2d-0.8.10.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-sports2d-0.8.10.dist-info/entry_points.txt,sha256=V8dFDIXatz9VvoGgoHzb2wE71C9-f85K6_OjnEQlxww,108
-sports2d-0.8.10.dist-info/top_level.txt,sha256=cWWBiDD2WbQXMoIoN6-9et9U2t2c_ZKo2JtBqO5uN-k,17
-sports2d-0.8.10.dist-info/RECORD,,
+sports2d-0.8.12.dist-info/licenses/LICENSE,sha256=f4qe3nE0Y7ltJho5w-xAR0jI5PUox5Xl-MsYiY7ZRM8,1521
+sports2d-0.8.12.dist-info/METADATA,sha256=g-tjMxMWMTSwStQfCCrmr65XPFeXIv37OO3XG6MMWxY,38443
+sports2d-0.8.12.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+sports2d-0.8.12.dist-info/entry_points.txt,sha256=V8dFDIXatz9VvoGgoHzb2wE71C9-f85K6_OjnEQlxww,108
+sports2d-0.8.12.dist-info/top_level.txt,sha256=cWWBiDD2WbQXMoIoN6-9et9U2t2c_ZKo2JtBqO5uN-k,17
+sports2d-0.8.12.dist-info/RECORD,,