This is an awiros utility to convert all your yolo models into the common onnx format.
Steps:
- Create a docker container with the following docker image: nvcr.io/nvidia/pytorch:21.08-py3
- In your workspace, clone this repo using git clone https://github.com/nachi-hebbar/yolo-to-onnx
- Install dependencies using pip install -r requirements.txt
- Run the onnx-converter.sh script
- Enter your model information as and when prompted by the onnx_converter script