From 8b0d87ff815b7ac6370e6123b6344c2a9a12fd22 Mon Sep 17 00:00:00 2001 From: Michael Welter 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 resolution { b, "force-resolution", 0 }; value deadzone_size { b, "deadzone-size", 1. }; value deadzone_hardness { b, "deadzone-hardness", 1.5 }; + value 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_; @@ -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