From 3a82ecde8a6efb748430caabf87701fef0eb46bc Mon Sep 17 00:00:00 2001 From: SzabolcsGergely Date: Fri, 23 Sep 2022 21:23:55 +0300 Subject: [PATCH] Set Yolo default iou threshold to 0.5 --- src/pipeline/node/DetectionNetwork.cpp | 1 + 1 file changed, 1 insertion(+) diff --git a/src/pipeline/node/DetectionNetwork.cpp b/src/pipeline/node/DetectionNetwork.cpp index 78a639a5d..4176182f4 100644 --- a/src/pipeline/node/DetectionNetwork.cpp +++ b/src/pipeline/node/DetectionNetwork.cpp @@ -47,6 +47,7 @@ YoloDetectionNetwork::YoloDetectionNetwork(const std::shared_ptr& YoloDetectionNetwork::YoloDetectionNetwork(const std::shared_ptr& par, int64_t nodeId, std::unique_ptr props) : NodeCRTP(par, nodeId, std::move(props)) { properties.parser.nnFamily = DetectionNetworkType::YOLO; + properties.parser.iouThreshold = 0.5f; } void YoloDetectionNetwork::setNumClasses(const int numClasses) {