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

Batching of poses in cpp implementation.

Browse Source
This commit is contained in:
Daniel
2025-01-20 12:20:25 +01:00
parent 8f2322694a
commit 86d8ccf797
5 changed files with 206 additions and 117 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
ros/rpt2D_wrapper_py/rpt2D_wrapper_py/rpt2D_wrapper.py
View File

@ -37,7 +37,7 @@ stop_flag = False
# Model config
min_bbox_score = 0.4
min_bbox_area = 0.1 * 0.1
batch_poses = False
batch_poses = True
# ==================================================================================================
@ -169,7 +169,9 @@ def main():
# Load 2D pose model
whole_body = test_triangulate.whole_body
if any((whole_body[k] for k in whole_body)):
kpt_model = utils_2d_pose.load_wb_model()
kpt_model = utils_2d_pose.load_wb_model(
min_bbox_score, min_bbox_area, batch_poses
)
else:
kpt_model = utils_2d_pose.load_model(min_bbox_score, min_bbox_area, batch_poses)