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

synchronizes on person with highest mean confidence

Browse Source
This commit is contained in:
davidpagnon 2024-09-17 10:25:19 +02:00
parent 7248c8a582
commit 394a162d76
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
Pose2Sim/synchronization.py
View File

@ -82,7 +82,9 @@ def convert_json2pandas(json_files, likelihood_threshold=0.6):
for j_p in json_files: for j_p in json_files:
with open(j_p) as j_f: with open(j_p) as j_f:
try: try:
json_data = json.load(j_f)['people'][0]['pose_keypoints_2d'] json_data = json.load(j_f)['people']
max_confidence_person = max(json_data, key=lambda p: np.mean(p['pose_keypoints_2d'][2::3]))
json_data = max_confidence_person['pose_keypoints_2d']
# remove points with low confidence # remove points with low confidence
json_data = np.array([[json_data[3*i],json_data[3*i+1],json_data[3*i+2]] if json_data[3*i+2]>likelihood_threshold else [0.,0.,0.] for i in range(nb_coord)]).ravel().tolist() json_data = np.array([[json_data[3*i],json_data[3*i+1],json_data[3*i+2]] if json_data[3*i+2]>likelihood_threshold else [0.,0.,0.] for i in range(nb_coord)]).ravel().tolist()
except: except: