I have referred to this link below which suggests trimming the batchedNMS branch and using the NMS plugin later. Is this the best practice? [image] Errors while reading ONNX file produced by TAO 5 TAO Toolkit It seems like the ONNX model contains an unsupported…

[image] venkatesha2411: How do we edit yolov4 export.py code in colab As mentioned above, you can change code via sed, see Unable to export hdf5 to etlt after Tao Training on Colab - #11 by Morganh . But it’s not convenient. Suggest you to modify code locally, then copy into google drive.

[image] venkatesha2411: Can we change the yolo_export code in the backend without NMS branch while converting yolov3/v4 model to onnx Yes, you can also change the export code. [image] venkatesha2411: how do we do it on colab? It is not very convenient in colab. But you can still refer…

How do we edit yolov4 export.py code in colab to convert the hdf5 model to onnx without batchedNMS branch instead of trimming the model?

Converting yolov3 and yolov4 model to onnx without batchedNMS

Accelerated Computing Intelligent Video Analytics TAO Toolkit

Morganh May 28, 2024, 2:42am 3

Yes, you can also change the export code.

It is not very convenient in colab. But you can still refer to Unable to export hdf5 to etlt after Tao Training on Colab - #9 by Morganh and Unable to export hdf5 to etlt after Tao Training on Colab - #11 by Morganh about how to login docker and modify code.

Topic		Replies	Views
YoloV3 without Batched NMS post processing code TAO Toolkit	2	187	October 21, 2024
Do we have to use BatchedNMSDynamic_TRT in the model output by ONNX? TAO Toolkit onnx	1	654	October 26, 2023
Errors while reading ONNX file produced by TAO 5 TAO Toolkit	10	2325	September 6, 2023
Adding BatchedNMSDynamic_TRT plugin in the ssd mobileNet onnx model TensorRT	4	2032	September 7, 2021
Yolov4 on tao gives batchedNMS error TAO Toolkit	4	453	February 12, 2024
My Yolov4 tf2checkpoint->save_model->onnx->trt TensorRT tensorrt , jetson-inference	3	584	November 1, 2021
Inference of tao trained yolov3 model(kitti format) using onnx runtime on my pc TAO Toolkit ai-training , tao	2	144	November 1, 2024
Onnx yolov3 postprocessing TAO Toolkit	5	295	November 1, 2024
Yolo TAO non maximum suppresion (NMS) TAO Toolkit yolo , tao	3	718	December 9, 2022
Make tiny yolo v4 onnx output compatible with nms layer batchedNMSPlugin (C++/TensoRT) TensorRT	2	1697	November 24, 2021

Converting yolov3 and yolov4 model to onnx without batchedNMS

Related topics