use_simple_model argument (faster but no muscles or flexible spine)

Author: davidpagnon

README.md

@@ -446,7 +446,7 @@ sports2d --help
 'do_ik': ["", "do inverse kinematics. false if not specified"],
 'use_augmentation': ["", "Use LSTM marker augmentation. false if not specified"],
 'feet_on_floor': ["", "offset marker augmentation results so that feet are at floor level. true if not specified"],
-'use_contacts_muscles': ["", "Use model with contact spheres and muscles. false if not specified"],
+'use_simple_model': ["", "IK 10+ times faster, but no muscles or flexible spine. false if not specified"],
 'participant_mass': ["", "mass of the participant in kg or none. Defaults to 70 if not provided. No influence on kinematics (motion), only on kinetics (forces)"],
 'close_to_zero_speed_m': ["","Sum for all keypoints: about 50 px/frame or 0.2 m/frame"], 
 'tracking_mode': ["", "'sports2d' or 'deepsort'. 'deepsort' is slower, harder to parametrize but can be more robust if correctly tuned"],

Sports2D/Demo/Config_demo.toml

@@ -152,7 +152,7 @@ filter_type = 'butterworth'   # butterworth, gaussian, LOESS, median
 do_ik = false # Do scaling and inverse kinematics?
 use_augmentation = false  # true or false (lowercase) # Set to true if you want to use the model with augmented markers
 feet_on_floor = false # true or false (lowercase) # Set to false if you want to use the model with feet not on the floor (e.g. running, jumping, etc.)
-use_contacts_muscles = true # true or false (lowercase) # If true, contact spheres and muscles are added to the model
+use_simple_model = false # true or false # >10 times faster IK if true. No muscles, no constraints (eg stiff spine and shoulders, no patella)
 participant_mass = [55.0, 67.0] # kg # defaults to 70 if not provided. No influence on kinematics (motion), only on kinetics (forces)
 right_left_symmetry = true # true or false (lowercase) # Set to false only if you have good reasons to think the participant is not symmetrical (e.g. prosthetic limb)
 

Sports2D/Sports2D.py

@@ -243,7 +243,7 @@
                     'kinematics':{'do_ik': False,
                                           'use_augmentation': False,
                                           'feet_on_floor': False,
-                                          'use_contacts_muscles': True,
+                                          'use_simple_model ': True,
                                           'participant_mass': [55.0, 67.0], 
                                           'right_left_symmetry': True,
                                           'default_height': 1.70,
@@ -297,7 +297,7 @@
                 'do_ik': ["", "do inverse kinematics. false if not specified"],
                 'use_augmentation': ["", "Use LSTM marker augmentation. false if not specified"],
                 'feet_on_floor': ["", "offset marker augmentation results so that feet are at floor level. true if not specified"],
-                'use_contacts_muscles': ["", "Use model with contact spheres and muscles. false if not specified"],
+                'use_simple_model ': ["", ">10 times faster IK if true. No muscles, no constraints (eg stiff spine and shoulders, no patella). false if not specified"],
                 'participant_mass': ["", "mass of the participant in kg or none. Defaults to 70 if not provided. No influence on kinematics (motion), only on kinetics (forces)"],
                 'close_to_zero_speed_m': ["","Sum for all keypoints: about 50 px/frame or 0.2 m/frame"], 
                 'tracking_mode': ["", "'sports2d' or 'deepsort'. 'deepsort' is slower, harder to parametrize but can be more robust if correctly tuned"],

Sports2D/process.py

@@ -1735,7 +1735,6 @@ def process_fun(config_dict, video_file, time_range, frame_rate, result_dir):
     all_frames_scores_homog = all_frames_scores_homog[...,new_keypoints_ids]
 
     frame_range = [0,frame_count] if video_file == 'webcam' else frame_range
-    print(frame_range)
     all_frames_time = pd.Series(np.linspace(frame_range[0]/fps, frame_range[1]/fps, frame_count-frame_range[0]), name='time')
     if load_trc_px:
         selected_persons = [0]