tensorrt_yolov4_ros icon indicating copy to clipboard operation
tensorrt_yolov4_ros copied to clipboard

Published 20 hours ago • verified publisher icon wep21

How to convert it to onnx in ros？

Open fy123-gh opened this issue 2 years ago • 1 comments

Hello, I am currently working on yolo->onnx->tensorrt and deploying to ROS, can you give me some reference?

fy123-gh avatar May 13 '22 03:05 fy123-gh

@fy123-gh You can use tihs script for yolov3. https://github.com/NVIDIA/TensorRT/blob/main/samples/python/yolov3_onnx/yolov3_to_onnx.py If you also want to convert yolov4 model, you need to add some patches into the script above.

wep21 avatar May 22 '22 09:05 wep21