From ae4f05419e93c7c382fae8fbca57295343e676a3 Mon Sep 17 00:00:00 2001 From: Michael Welter Date: Sat, 2 Nov 2024 12:08:25 +0100 Subject: tracker/nn: Update models * Remove head-pose.onnx * Add head-pose-0.3-big-quantized.onnx, which was trained with https://github.com/opentrack/neuralnet-tracker-traincode commit 9891d60 --- tracker-neuralnet/CMakeLists.txt | 2 +- tracker-neuralnet/ftnoir_tracker_neuralnet.h | 2 +- .../models/head-pose-0.3-big-quantized.onnx | Bin 0 -> 11385815 bytes tracker-neuralnet/models/head-pose.onnx | Bin 13047683 -> 0 bytes 4 files changed, 2 insertions(+), 2 deletions(-) create mode 100644 tracker-neuralnet/models/head-pose-0.3-big-quantized.onnx delete mode 100644 tracker-neuralnet/models/head-pose.onnx (limited to 'tracker-neuralnet') diff --git a/tracker-neuralnet/CMakeLists.txt b/tracker-neuralnet/CMakeLists.txt index e7bb3b1c..3729c789 100644 --- a/tracker-neuralnet/CMakeLists.txt +++ b/tracker-neuralnet/CMakeLists.txt @@ -33,9 +33,9 @@ if(OpenCV_FOUND AND ONNXRuntime_FOUND AND OpenMP_FOUND) install( FILES "models/head-localizer.onnx" - "models/head-pose.onnx" "models/head-pose-0.2-big.onnx" "models/head-pose-0.2-small.onnx" + "models/head-pose-0.3-big-quantized.onnx" DESTINATION "${opentrack-libexec}/models" PERMISSIONS ${opentrack-perms-file} ) diff --git a/tracker-neuralnet/ftnoir_tracker_neuralnet.h b/tracker-neuralnet/ftnoir_tracker_neuralnet.h index bafaa6e5..60715da3 100644 --- a/tracker-neuralnet/ftnoir_tracker_neuralnet.h +++ b/tracker-neuralnet/ftnoir_tracker_neuralnet.h @@ -84,7 +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" }; + value posenet_file { b, "posenet-file", "head-pose-0.3-big-quantized.onnx" }; Settings(); }; diff --git a/tracker-neuralnet/models/head-pose-0.3-big-quantized.onnx b/tracker-neuralnet/models/head-pose-0.3-big-quantized.onnx new file mode 100644 index 00000000..7f875c63 Binary files /dev/null and b/tracker-neuralnet/models/head-pose-0.3-big-quantized.onnx differ diff --git a/tracker-neuralnet/models/head-pose.onnx b/tracker-neuralnet/models/head-pose.onnx deleted file mode 100644 index dcb55dcc..00000000 Binary files a/tracker-neuralnet/models/head-pose.onnx and /dev/null differ -- cgit v1.2.3