sports2d 0.8.9__py3-none-any.whl → 0.8.11__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- Sports2D/Sports2D.py +2 -2
- Sports2D/Utilities/tests.py +2 -2
- Sports2D/process.py +10 -4
- {sports2d-0.8.9.dist-info → sports2d-0.8.11.dist-info}/METADATA +1 -1
- {sports2d-0.8.9.dist-info → sports2d-0.8.11.dist-info}/RECORD +9 -9
- {sports2d-0.8.9.dist-info → sports2d-0.8.11.dist-info}/WHEEL +0 -0
- {sports2d-0.8.9.dist-info → sports2d-0.8.11.dist-info}/entry_points.txt +0 -0
- {sports2d-0.8.9.dist-info → sports2d-0.8.11.dist-info}/licenses/LICENSE +0 -0
- {sports2d-0.8.9.dist-info → sports2d-0.8.11.dist-info}/top_level.txt +0 -0
Sports2D/Sports2D.py
CHANGED
|
@@ -243,7 +243,7 @@ DEFAULT_CONFIG = {'base': {'video_input': ['demo.mp4'],
|
|
|
243
243
|
'kinematics':{'do_ik': False,
|
|
244
244
|
'use_augmentation': False,
|
|
245
245
|
'feet_on_floor': False,
|
|
246
|
-
'use_simple_model
|
|
246
|
+
'use_simple_model': False,
|
|
247
247
|
'participant_mass': [55.0, 67.0],
|
|
248
248
|
'right_left_symmetry': True,
|
|
249
249
|
'default_height': 1.70,
|
|
@@ -297,7 +297,7 @@ CONFIG_HELP = {'config': ["C", "path to a toml configuration file"],
|
|
|
297
297
|
'do_ik': ["", "do inverse kinematics. false if not specified"],
|
|
298
298
|
'use_augmentation': ["", "Use LSTM marker augmentation. false if not specified"],
|
|
299
299
|
'feet_on_floor': ["", "offset marker augmentation results so that feet are at floor level. true if not specified"],
|
|
300
|
-
'use_simple_model
|
|
300
|
+
'use_simple_model': ["", ">10 times faster IK if true. No muscles, no constraints (eg stiff spine and shoulders, no patella). false if not specified"],
|
|
301
301
|
'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)"],
|
|
302
302
|
'close_to_zero_speed_m': ["","Sum for all keypoints: about 50 px/frame or 0.2 m/frame"],
|
|
303
303
|
'tracking_mode': ["", "'sports2d' or 'deepsort'. 'deepsort' is slower, harder to parametrize but can be more robust if correctly tuned"],
|
Sports2D/Utilities/tests.py
CHANGED
|
@@ -97,8 +97,8 @@ def test_workflow():
|
|
|
97
97
|
config_dict.get("base").update({"video_dir": str(video_dir)})
|
|
98
98
|
config_dict.get("base").update({"person_ordering_method": "highest_likelihood"})
|
|
99
99
|
with open(config_path, 'w') as f: toml.dump(config_dict, f)
|
|
100
|
-
|
|
101
|
-
subprocess.run(
|
|
100
|
+
demo_cmd5 = ["sports2d", "--config", str(config_path), "--show_realtime_results", "False", "--show_graphs", "False"]
|
|
101
|
+
subprocess.run(demo_cmd5, check=True, capture_output=True, text=True, encoding='utf-8')
|
|
102
102
|
|
|
103
103
|
|
|
104
104
|
if __name__ == "__main__":
|
Sports2D/process.py
CHANGED
|
@@ -60,6 +60,7 @@ import copy
|
|
|
60
60
|
import shutil
|
|
61
61
|
import os
|
|
62
62
|
import re
|
|
63
|
+
import platform
|
|
63
64
|
from importlib.metadata import version
|
|
64
65
|
from datetime import datetime
|
|
65
66
|
import itertools as it
|
|
@@ -788,18 +789,23 @@ def pose_plots(trc_data_unfiltered, trc_data, person_id):
|
|
|
788
789
|
OUTPUT:
|
|
789
790
|
- matplotlib window with tabbed figures for each keypoint
|
|
790
791
|
'''
|
|
791
|
-
|
|
792
|
-
|
|
792
|
+
os_name = platform.system()
|
|
793
|
+
|
|
794
|
+
if os_name == 'Windows':
|
|
795
|
+
mpl.use('qt5agg') # windows
|
|
793
796
|
mpl.rc('figure', max_open_warning=0)
|
|
794
797
|
|
|
795
798
|
keypoints_names = trc_data.columns[1::3]
|
|
796
|
-
|
|
799
|
+
|
|
797
800
|
pw = plotWindow()
|
|
798
801
|
pw.MainWindow.setWindowTitle('Person'+ str(person_id) + ' coordinates') # Main title
|
|
799
802
|
|
|
800
803
|
for id, keypoint in enumerate(keypoints_names):
|
|
801
804
|
f = plt.figure()
|
|
802
|
-
|
|
805
|
+
if os_name == 'Windows':
|
|
806
|
+
f.canvas.manager.window.setWindowTitle(keypoint + ' Plot') # windows
|
|
807
|
+
elif os_name == 'Darwin': # macOS
|
|
808
|
+
f.canvas.manager.set_window_title(keypoint + ' Plot') # mac
|
|
803
809
|
|
|
804
810
|
axX = plt.subplot(211)
|
|
805
811
|
plt.plot(trc_data_unfiltered.iloc[:,0], trc_data_unfiltered.iloc[:,id*3+1], label='unfiltered')
|
|
@@ -9,18 +9,18 @@ Content/paper.md,sha256=8rWSOLrKTysloZv0Fz2lr3nayxtHi7GlFMqwdgDVggY,11333
|
|
|
9
9
|
Content/sports2d_blender.gif,sha256=wgMuPRxhja3XtQn76_okGXsNnUT9Thp0pnD36GdW5_E,448786
|
|
10
10
|
Content/sports2d_opensim.gif,sha256=XP1AcjqhbGcJknXUoNJjPWAwaM9ahZafbDgLWvzKJs4,376656
|
|
11
11
|
Sports2D/Sports2D.ipynb,sha256=VnOVjIl6ndnCJTT13L4W5qTw4T-TQDF3jt3-wxnXDqM,2427047
|
|
12
|
-
Sports2D/Sports2D.py,sha256=
|
|
12
|
+
Sports2D/Sports2D.py,sha256=tbQi_d6GXMqFkSd1_FRxyl6oA4CXEJH0g-lEskmR4mI,33521
|
|
13
13
|
Sports2D/__init__.py,sha256=BuUkPEdItxlkeqz4dmoiPwZLkgAfABJK3KWQ1ujTGwE,153
|
|
14
|
-
Sports2D/process.py,sha256=
|
|
14
|
+
Sports2D/process.py,sha256=pcPW5qpZzD-L-ZiRFZn3TWeywg3ojlgjhiw2FPkGodA,111598
|
|
15
15
|
Sports2D/Demo/Config_demo.toml,sha256=jiPGr6JbePu1tItgRxdAu45mwFqr3RcIi_tfhO2ik1g,14027
|
|
16
16
|
Sports2D/Demo/demo.mp4,sha256=2aZkFxhWR7ESMEtXCT8MGA83p2jmoU2sp1ylQfO3gDk,3968304
|
|
17
17
|
Sports2D/Utilities/__init__.py,sha256=BuUkPEdItxlkeqz4dmoiPwZLkgAfABJK3KWQ1ujTGwE,153
|
|
18
18
|
Sports2D/Utilities/common.py,sha256=idMRmesFv5BPX-5g3z5dOVa7SpS_8tNgijvGrOZlR-k,11185
|
|
19
19
|
Sports2D/Utilities/filter.py,sha256=rfZcqofjllKI_5ovZTKEAmyjOZpB_PzbAJ0P874T8Ak,4973
|
|
20
|
-
Sports2D/Utilities/tests.py,sha256=
|
|
21
|
-
sports2d-0.8.
|
|
22
|
-
sports2d-0.8.
|
|
23
|
-
sports2d-0.8.
|
|
24
|
-
sports2d-0.8.
|
|
25
|
-
sports2d-0.8.
|
|
26
|
-
sports2d-0.8.
|
|
20
|
+
Sports2D/Utilities/tests.py,sha256=3-slY2teKd78q0NzfJ2H84xRFSNXvGyDwew27oHJfrY,4740
|
|
21
|
+
sports2d-0.8.11.dist-info/licenses/LICENSE,sha256=f4qe3nE0Y7ltJho5w-xAR0jI5PUox5Xl-MsYiY7ZRM8,1521
|
|
22
|
+
sports2d-0.8.11.dist-info/METADATA,sha256=rKNwu5yaVBb4c161s5SFrGYxyJWoQdqcyzPn5YxEwGA,38443
|
|
23
|
+
sports2d-0.8.11.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
|
|
24
|
+
sports2d-0.8.11.dist-info/entry_points.txt,sha256=V8dFDIXatz9VvoGgoHzb2wE71C9-f85K6_OjnEQlxww,108
|
|
25
|
+
sports2d-0.8.11.dist-info/top_level.txt,sha256=cWWBiDD2WbQXMoIoN6-9et9U2t2c_ZKo2JtBqO5uN-k,17
|
|
26
|
+
sports2d-0.8.11.dist-info/RECORD,,
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|