From 8b0d87ff815b7ac6370e6123b6344c2a9a12fd22 Mon Sep 17 00:00:00 2001
From: Michael Welter <michael@welter-4d.de>
Date: Sun, 1 Oct 2023 14:39:10 +0200
Subject: tracker/nn: Add button to choose pose net file

---
 tracker-neuralnet/ftnoir_tracker_neuralnet.h | 3 +++
 1 file changed, 3 insertions(+)

(limited to 'tracker-neuralnet/ftnoir_tracker_neuralnet.h')

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();
 };
 
 
-- 
cgit v1.2.3