Display image with correct ratio + more info on angles in Readme

Author: davidpagnon

README.md

@@ -364,6 +364,11 @@ sports2d --video_input demo.mp4 other_video.mp4 --time_range 1.2 2.7 0 3.5
   ```cmd
   sports2d --flip_left_right true # Default
   ```
+- Correct segment angles according to the estimated camera tild angle.\
+  **N.B.:** *The camera tilt angle is automatically estimated. Set to false if it is actually the floor which is tilted rather than the camera.*
+  ```cmd
+  sports2d --correct_segment_angles_with_floor_angle true # Default
+  ```
 
 - To run **inverse kinematics with OpenSim**, check [this section](#run-inverse-kinematics)
 
@@ -469,7 +474,7 @@ sports2d --help
 'trimmed_extrema_percent': ["", "Proportion of the most extreme segment values to remove before calculating their mean. Defaults to 50"],
 'fontSize': ["", "font size for angle values. 0.3 if not specified"],
 'flip_left_right': ["", "true or false. Flips angles when the person faces the other side. The person looks to the right if their toe keypoint is to the right of their heel. Set it to false if the person is sprinting or if you want timeseries to be continuous even when the participant switches their stance. true if not specified"],
-'correct_segment_angles_with_floor_angle': ["", "true or false. If the camera is tilted, corrects segment angles as regards to the floor angle. Set to false is the floor is tilted instead. True if not specified"],
+'correct_segment_angles_with_floor_angle': ["", "true or false. If the camera is tilted, corrects segment angles as regards to the floor angle. Set to false if it is actually the floor which is tilted, not the camera. True if not specified"],
 'interpolate': ["", "interpolate missing data. true if not specified"],
 'interp_gap_smaller_than': ["", "interpolate sequences of missing data if they are less than N frames long. 10 if not specified"],
 'fill_large_gaps_with': ["", "last_value, nan, or zeros. last_value if not specified"],

Sports2D/process.py

@@ -1512,8 +1512,13 @@ def process_fun(config_dict, video_file, time_range, frame_rate, result_dir):
         frame_range = [int((time_range[0]-start_time) * frame_rate), int((time_range[1]-start_time) * frame_rate)] if time_range else [0, int(cap.get(cv2.CAP_PROP_FRAME_COUNT))]
         frame_iterator = tqdm(range(*frame_range)) # use a progress bar
     if show_realtime_results:
-        cv2.namedWindow(f'{video_file} Sports2D', cv2.WINDOW_NORMAL + cv2.WINDOW_KEEPRATIO)
-        cv2.setWindowProperty(f'{video_file} Sports2D', cv2.WND_PROP_ASPECT_RATIO, cv2.WINDOW_FULLSCREEN)
+        try: 
+            screen_width, screen_height = get_screen_size()
+            display_width, display_height = calculate_display_size(cam_width, cam_height, screen_width, screen_height)
+            cv2.namedWindow(f'{video_file} Sports2D', cv2.WINDOW_NORMAL)
+            cv2.resizeWindow(f'{video_file} Sports2D', display_width, display_height)
+        except: # if Pose2Sim < v0.10.29
+            cv2.namedWindow(f'{video_file} Sports2D', cv2.WINDOW_NORMAL + cv2.WINDOW_KEEPRATIO)
 
     # Select the appropriate model based on the model_type
     logging.info('\nEstimating pose...')