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

Improved tracker parametrization.

Browse Source
This commit is contained in:
Daniel
2025-04-23 10:40:59 +02:00
parent 98399cc00e
commit cc8b75b8bf
3 changed files with 59 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
scripts/test_skelda_dataset.cpp
View File

@ -123,6 +123,8 @@ int main(int argc, char **argv)
const size_t min_group_size = config["min_group_size"];
const int take_interval = config["take_interval"];
const float ifps = config["fps"];
const float max_movement_speed = config["max_movement_speed"];
const float max_track_distance = config["max_track_distance"];
// Load 2D model
bool use_wb = utils_pipeline::use_whole_body(whole_body);
@ -134,7 +136,7 @@ int main(int argc, char **argv)
std::unique_ptr<Triangulator> tri_model = std::make_unique<Triangulator>(
min_match_score, min_group_size);
std::unique_ptr<PoseTracker> pose_tracker = std::make_unique<PoseTracker>(
ifps);
max_movement_speed, max_track_distance);
// Timers
size_t time_count = dataset.size();
@ -216,7 +218,7 @@ int main(int argc, char **argv)
auto &item = dataset[i];
auto &poses_2d = all_poses_2d[i];
if (old_scene != item["scene"] || old_id + take_interval < item["index"])
if (old_scene != item["scene"] || old_id + take_interval != item["index"])
{
// Reset last poses if scene changes
tri_model->reset();
@ -248,7 +250,7 @@ int main(int argc, char **argv)
elapsed = std::chrono::high_resolution_clock::now() - stime;
times_pose3d.push_back(elapsed.count());
if (ifps < 1)
if (ifps <= 0)
{
// Disable pose tracking if frame rate is too low
times_tracks.push_back(0.0);