fulyvv/pose2sim
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge branch 'main' of https://github.com/perfanalytics/pose2sim

Browse Source
This commit is contained in:
davidpagnon 2023-01-16 00:15:33 +01:00
commit 9b8fcae215
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
Pose2Sim/Utilities/DLC_to_OpenPose.py
View File

@ -84,7 +84,7 @@ def DLC_to_OpenPose_func(*args):
for f, frame in enumerate(h5_file.index): for f, frame in enumerate(h5_file.index):
h5_line = np.array([[h5_file.iloc[f, 3*k], h5_file.iloc[f, 3*k+1], h5_file.iloc[f, 3*k+2]] for k in range(kpt_nb)]).flatten().tolist() h5_line = np.array([[h5_file.iloc[f, 3*k], h5_file.iloc[f, 3*k+1], h5_file.iloc[f, 3*k+2]] for k in range(kpt_nb)]).flatten().tolist()
json_dict['people'][0]['pose_keypoints_2d'] = h5_line json_dict['people'][0]['pose_keypoints_2d'] = h5_line
json_file = os.path.join(json_folder_path, os.path.splitext(os.path.basename(str(frame)))[0]+'.json') json_file = os.path.join(json_folder_path, os.path.splitext(os.path.basename(str(frame).zfill(5)))[0]+'.json')
with open(json_file, 'w') as js_f: with open(json_file, 'w') as js_f:
js_f.write(json.dumps(json_dict)) js_f.write(json.dumps(json_dict))