Giters

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

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Error in ONNX export file

JoelDuring opened this issue · comments

commented

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.