start at frame 1 instead of 0

This commit is contained in:
David PAGNON 2024-07-23 00:14:15 +02:00 committed by GitHub
parent 1a8253aa24
commit 90d0ccc14a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -86,7 +86,7 @@ def get_marker_positions(motion_data, model, in_degrees=True, marker_list=[]):
marker_positions += [np.array([marker_set.get(mk_name).findLocationInFrame(state, model.getGround()).to_numpy() for mk_name in marker_set_names]).flatten()] marker_positions += [np.array([marker_set.get(mk_name).findLocationInFrame(state, model.getGround()).to_numpy() for mk_name in marker_set_names]).flatten()]
marker_positions_pd = pd.DataFrame(marker_positions, columns=marker_set_names_xyz) marker_positions_pd = pd.DataFrame(marker_positions, columns=marker_set_names_xyz)
marker_positions_pd.insert(0, 'time', times) marker_positions_pd.insert(0, 'time', times)
marker_positions_pd.insert(0, 'frame', np.arange(len(times))) marker_positions_pd.insert(0, 'frame', np.arange(len(times))+1)
return marker_positions_pd, marker_set_names return marker_positions_pd, marker_set_names