tracker/nn: Add button to choose pose net file

1 files changed, 3 insertions, 0 deletions

diff --git a/tracker-neuralnet/ftnoir_tracker_neuralnet.h b/tracker-neuralnet/ftnoir_tracker_neuralnet.h
index 3548335e..bafaa6e5 100644
--- a/tracker-neuralnet/ftnoir_tracker_neuralnet.h
+++ b/tracker-neuralnet/ftnoir_tracker_neuralnet.h
@@ -84,6 +84,7 @@ struct Settings : opts {
     value<int> resolution { b, "force-resolution", 0 };
     value<double> deadzone_size { b, "deadzone-size", 1. };
     value<double> deadzone_hardness { b, "deadzone-hardness", 1.5 };
+    value<QString> posenet_file { b, "posenet-file", "head-pose.onnx" };
     Settings();
 };
 
@@ -126,6 +127,7 @@ private:
     QuatPose compute_filtered_pose(const PoseEstimator::Face &face);
     // Compute the pose in 3d space taking the network outputs
     QuatPose transform_to_world_pose(const cv::Quatf &face_rotation, const cv::Point2f& face_xy, const float face_size) const;
+    QString get_posenet_filename() const;
 
     Settings settings_;
     std::optional<Localizer> localizer_;
@@ -192,6 +194,7 @@ private Q_SLOTS:
     void startstop_trans_calib(bool start);
     void trans_calib_step();
     void status_poll();
+    void onSelectPoseNetFile();
 };