crosstyan/RapidPoseTriangulation
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Some more performance improvements.

Browse Source
This commit is contained in:
Daniel
2024-07-03 11:02:02 +02:00
parent 089fb0b41d
commit 3fcca51d17
4 changed files with 1238 additions and 1236 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2462
media/RESULTS.md
View File

File diff suppressed because it is too large Load Diff

6
scripts/test_skelda_dataset.py
View File

@ -356,10 +356,12 @@ def main():
# Choose this depending on the fraction of invalid/missing persons # Choose this depending on the fraction of invalid/missing persons
# A higher value reduces the number of proposals # A higher value reduces the number of proposals
"panoptic": 0.95, "panoptic": 0.95,
"human36m": 0.96, "human36m": 0.94,
"mvor": 0.87, "mvor": 0.86,
"campus": 0.96, "campus": 0.96,
"shelf": 0.96, "shelf": 0.96,
"ikeaasm": 0.86,
"tsinghua": 0.96,
} }
minscore = minscores.get(dataset_use, 0.95) minscore = minscores.get(dataset_use, 0.95)

2
scripts/test_triangulate.py
View File

@ -242,7 +242,7 @@ def filter_poses(poses3D, poses2D, roomparams, joint_names, drop_few_limbs=True)
if any(((d > 2.3) for d in diff)): if any(((d > 2.3) for d in diff)):
drop.append(i) drop.append(i)
continue continue
if all(((d < 0.4) for d in diff)): if all(((d < 0.3) for d in diff)):
drop.append(i) drop.append(i)
continue continue
if ( if (

4
scripts/utils_2d_pose.py
View File

@ -14,8 +14,8 @@ def load_model():
print("Loading mmpose model ...") print("Loading mmpose model ...")
model = MMPoseInferencer( model = MMPoseInferencer(
pose2d="/mmpose/projects/rtmpose/rtmpose/body_2d_keypoint/rtmpose-l_8xb256-420e_coco-384x288.py", pose2d="/mmpose/projects/rtmpose/rtmpose/body_2d_keypoint/rtmpose-m_8xb256-420e_coco-384x288.py",
pose2d_weights="https://download.openmmlab.com/mmpose/v1/projects/rtmposev1/rtmpose-l_simcc-body7_pt-body7_420e-384x288-3f5a1437_20230504.pth", pose2d_weights="https://download.openmmlab.com/mmpose/v1/projects/rtmposev1/rtmpose-m_simcc-body7_pt-body7_420e-384x288-65e718c4_20230504.pth",
det_model="/mmpose/projects/rtmpose/rtmdet/person/rtmdet_nano_320-8xb32_coco-person.py", det_model="/mmpose/projects/rtmpose/rtmdet/person/rtmdet_nano_320-8xb32_coco-person.py",
det_weights="https://download.openmmlab.com/mmpose/v1/projects/rtmpose/rtmdet_nano_8xb32-100e_coco-obj365-person-05d8511e.pth", det_weights="https://download.openmmlab.com/mmpose/v1/projects/rtmpose/rtmdet_nano_8xb32-100e_coco-obj365-person-05d8511e.pth",
det_cat_ids=[0], det_cat_ids=[0],