marcoslucianops / DeepStream-Yolo-Seg

NVIDIA DeepStream SDK 6.3 / 6.2 / 6.1.1 / 6.1 / 6.0.1 / 6.0 implementation for YOLO-Segmentation models
MIT License
54 stars 12 forks source link

Error in ONNX export file #8

Open JoelDuring opened 1 year ago

JoelDuring commented 1 year ago

While tweaking are model I found an error in utils/export_yoloV8_seg.py. In line 71, the IOU threshold and confidence threshold are passed to the NMS.apply function in the wrong order. In our case, this causes the network to have very few detections when a high IOU threshold is set.