sports2d 0.8.15__tar.gz → 0.8.16__tar.gz

{sports2d-0.8.15 → sports2d-0.8.16}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sports2d
-Version: 0.8.15
+Version: 0.8.16
 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.15 → sports2d-0.8.16}/Sports2D/Utilities/tests.py

@@ -37,6 +37,7 @@ def test_workflow():
     Test the workflow of Sports2D.
     '''
 
+    from Sports2D import Sports2D
     root_dir = os.path.dirname(os.path.abspath(__file__))
     os.chdir(root_dir)
 
@@ -53,7 +54,6 @@ def test_workflow():
     config_dict.get("base").update({"show_realtime_results":False})
     config_dict.get("post-processing").update({"show_graphs":False})
     
-    from Sports2D import Sports2D
     Sports2D.process(config_dict)
 
 
@@ -82,7 +82,7 @@ def test_workflow():
                  "--pose_model", "body", "--mode", """{'pose_class':'RTMO', 'pose_model':'https://download.openmmlab.com/mmpose/v1/projects/rtmo/onnx_sdk/rtmo-m_16xb16-600e_body7-640x640-39e78cc4_20231211.zip', 'pose_input_size':[640, 640]}"""]
     subprocess.run(demo_cmd3, check=True, capture_output=True, text=True, encoding='utf-8')
     
-    # With a time range, inverse kinematics, marker augmentation, body pose_model and custom RTMO mode
+    # With a time range, inverse kinematics, marker augmentation
     demo_cmd4 = ["sports2d", "--person_ordering_method", "greatest_displacement", "--show_realtime_results", "False", "--show_graphs", "False",
                  "--time_range", "1.2", "2.7",
                  "--do_ik", "True", "--use_augmentation", "True", 

{sports2d-0.8.15 → sports2d-0.8.16}/Sports2D/process.py

@@ -1999,6 +1999,8 @@ def process_fun(config_dict, video_file, time_range, frame_rate, result_dir):
                 else:
                     visible_side_i = 'none'
                 new_visible_side += [visible_side_i]
+        else:
+            new_visible_side = visible_side.copy()
                 
             
 
@@ -2063,7 +2065,7 @@ def process_fun(config_dict, video_file, time_range, frame_rate, result_dir):
         for i, idx_person in enumerate(selected_persons):
             angles_path_person = angles_output_path.parent / (angles_output_path.stem + f'_person{i:02d}.mot')
             all_frames_angles_person = pd.DataFrame(all_frames_angles_homog[:,idx_person,:], columns=angle_names)
-            
+
             # Flip angles for left side when flip_left_right false
             if new_visible_side[i] == 'left' and not flip_left_right: 
                 all_frames_angles_homog[:, idx_person, :] = -all_frames_angles_homog[:, idx_person, :]
@@ -2169,19 +2171,22 @@ def process_fun(config_dict, video_file, time_range, frame_rate, result_dir):
         new_keypoints_ids = list(range(len(new_keypoints_ids)))
 
         # Draw pose and angles
+        if 'first_trim' not in locals():
+            first_trim, last_trim = 0, frame_count-1
         for frame_count, (frame, valid_X, valid_X_flipped, valid_Y, valid_scores, valid_angles) in enumerate(zip(frames, all_frames_X_processed, all_frames_X_flipped_processed, all_frames_Y_processed, all_frames_scores_processed, all_frames_angles_processed)):
-            img = frame.copy()
-            img = draw_bounding_box(img, valid_X, valid_Y, colors=colors, fontSize=fontSize, thickness=thickness)
-            img = draw_keypts(img, valid_X, valid_Y, valid_scores, cmap_str='RdYlGn')
-            img = draw_skel(img, valid_X, valid_Y, pose_model_with_new_ids)
-            if calculate_angles:
-                img = draw_angles(img, valid_X, valid_Y, valid_angles, valid_X_flipped, new_keypoints_ids, new_keypoints_names, angle_names, display_angle_values_on=display_angle_values_on, colors=colors, fontSize=fontSize, thickness=thickness)
-
-            # Save video or images
-            if save_vid:
-                out_vid.write(img)
-            if save_img:
-                cv2.imwrite(str((img_output_dir / f'{output_dir_name}_{(frame_count):06d}.png')), img)
+            if frame_count >= first_trim and frame_count <= last_trim:
+                img = frame.copy()
+                img = draw_bounding_box(img, valid_X, valid_Y, colors=colors, fontSize=fontSize, thickness=thickness)
+                img = draw_keypts(img, valid_X, valid_Y, valid_scores, cmap_str='RdYlGn')
+                img = draw_skel(img, valid_X, valid_Y, pose_model_with_new_ids)
+                if calculate_angles:
+                    img = draw_angles(img, valid_X, valid_Y, valid_angles, valid_X_flipped, new_keypoints_ids, new_keypoints_names, angle_names, display_angle_values_on=display_angle_values_on, colors=colors, fontSize=fontSize, thickness=thickness)
+
+                # Save video or images
+                if save_vid:
+                    out_vid.write(img)
+                if save_img:
+                    cv2.imwrite(str((img_output_dir / f'{output_dir_name}_{(frame_count+frame_range[0]):06d}.png')), img)
 
         if save_vid:
             out_vid.release()

{sports2d-0.8.15 → sports2d-0.8.16}/sports2d.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sports2d
-Version: 0.8.15
+Version: 0.8.16
 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>