diff options
Diffstat (limited to 'samples/ObjectDetection/include/YoloResultDecoder.hpp')
-rw-r--r-- | samples/ObjectDetection/include/YoloResultDecoder.hpp | 6 |
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; |