Merge pull request #596 from luxonis/yolo_iou_threshold

Set default yolo iou threshold to 0.5
luxonis · Sep 24, 2022 · 4f9a8e4 · 4f9a8e4
2 parents 875bd7b + 3a82ecd
commit 4f9a8e4
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/pipeline/node/DetectionNetwork.cpp b/src/pipeline/node/DetectionNetwork.cpp
@@ -47,6 +47,7 @@ YoloDetectionNetwork::YoloDetectionNetwork(const std::shared_ptr<PipelineImpl>&
 YoloDetectionNetwork::YoloDetectionNetwork(const std::shared_ptr<PipelineImpl>& par, int64_t nodeId, std::unique_ptr<Properties> props)
     : NodeCRTP<DetectionNetwork, YoloDetectionNetwork, DetectionNetworkProperties>(par, nodeId, std::move(props)) {
     properties.parser.nnFamily = DetectionNetworkType::YOLO;
+    properties.parser.iouThreshold = 0.5f;
 }
 
 void YoloDetectionNetwork::setNumClasses(const int numClasses) {