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

Changed detector for wb poses.

Browse Source
This commit is contained in:
Daniel
2025-01-27 15:46:55 +01:00
parent 45d003c2e9
commit d8beaecc56
4 changed files with 799 additions and 782 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
scripts/utils_2d_pose.hpp
View File

@ -1048,7 +1048,9 @@ namespace utils_2d_pose
bool batch_poses = false)
{
std::string base_path = "/RapidPoseTriangulation/extras/mmdeploy/exports/";
std::string path_det = base_path + "rtmdet-nano_1x320x320x3_fp16_extra-steps.onnx";
std::string path_det_n1 = base_path + "rtmdet-nano_1x320x320x3_fp16_extra-steps.onnx";
std::string path_det_m1 = base_path + "rtmdet-m_1x320x320x3_fp16_extra-steps.onnx";
std::string path_pose_m1 = base_path + "rtmpose-m_1x384x288x3_fp16_extra-steps.onnx";
std::string path_pose_mb = base_path + "rtmpose-m_Bx384x288x3_fp16_extra-steps.onnx";
@ -1056,9 +1058,11 @@ namespace utils_2d_pose
std::string path_pose_wb = base_path + "rtmpose-l_wb_Bx384x288x3_extra-steps.onnx";
this->num_joints = whole_body ? 133 : 17;
std::string path_det;
std::string path_pose;
if (!whole_body)
{
path_det = path_det_n1;
if (!batch_poses)
{
path_pose = path_pose_m1;
@ -1070,6 +1074,7 @@ namespace utils_2d_pose
}
else
{
path_det = path_det_m1;
if (!batch_poses)
{
path_pose = path_pose_w1;