I have referred to this link below which suggests trimming the batchedNMS branch and using the NMS plugin later. Is this the best practice?
Can we change the yolo_export code in the backend without NMS branch while converting yolov3/v4 model to onnx, how do we do it on colab?
1 post - 1 participant
Read full topic