aboutsummaryrefslogtreecommitdiff
path: root/samples/ObjectDetection/include/YoloResultDecoder.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'samples/ObjectDetection/include/YoloResultDecoder.hpp')
-rw-r--r--samples/ObjectDetection/include/YoloResultDecoder.hpp6
1 files changed, 3 insertions, 3 deletions
diff --git a/samples/ObjectDetection/include/YoloResultDecoder.hpp b/samples/ObjectDetection/include/YoloResultDecoder.hpp
index 98435e3cc9..ae6cb5e710 100644
--- a/samples/ObjectDetection/include/YoloResultDecoder.hpp
+++ b/samples/ObjectDetection/include/YoloResultDecoder.hpp
@@ -26,9 +26,9 @@ public:
*/
YoloResultDecoder(float NMSThreshold, float ClsThreshold, float ObjectThreshold);
- DetectedObjects Decode(const InferenceResults& results,
- const Size& outputFrameSize,
- const Size& resizedFrameSize,
+ DetectedObjects Decode(const common::InferenceResults<float>& results,
+ const common::Size& outputFrameSize,
+ const common::Size& resizedFrameSize,
const std::vector <std::string>& labels) override;
private:
float m_NmsThreshold;