summaryrefslogtreecommitdiffhomepage
path: root/tracker-neuralnet/ftnoir_tracker_neuralnet.h
diff options
context:
space:
mode:
authorStanislaw Halik <sthalik@misaki.pl>2024-02-11 15:01:26 +0100
committerStanislaw Halik <sthalik@misaki.pl>2024-02-11 15:01:26 +0100
commit47bc9e57ad55128ae0a5bf206f3931a3aa4f5dc1 (patch)
tree778714f2ef7925303375e38a4b6674cc9eae7b95 /tracker-neuralnet/ftnoir_tracker_neuralnet.h
parentd81fe9c5fd4fa54ecb821d627a030c6c1c81ed8a (diff)
tracker/nn: update from master
Diffstat (limited to 'tracker-neuralnet/ftnoir_tracker_neuralnet.h')
-rw-r--r--tracker-neuralnet/ftnoir_tracker_neuralnet.h5
1 files changed, 4 insertions, 1 deletions
diff --git a/tracker-neuralnet/ftnoir_tracker_neuralnet.h b/tracker-neuralnet/ftnoir_tracker_neuralnet.h
index d44a9953..e5696cc9 100644
--- a/tracker-neuralnet/ftnoir_tracker_neuralnet.h
+++ b/tracker-neuralnet/ftnoir_tracker_neuralnet.h
@@ -86,6 +86,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-0.2-small.onnx" };
Settings();
};
@@ -128,6 +129,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_;
@@ -146,7 +148,7 @@ private:
double fps_ = 0;
double inference_time_ = 0;
cv::Size resolution_ = {};
-
+
static constexpr double RC = .25;
int num_threads_ = 1;
bool is_visible_ = true;
@@ -196,6 +198,7 @@ private Q_SLOTS:
void startstop_trans_calib(bool start);
void trans_calib_step();
void status_poll();
+ void onSelectPoseNetFile();
};