Skip to content

nachi-hebbar/yolo-to-onnx

Repository files navigation

yolo-to-onnx

This is an awiros utility to convert all your yolo models into the common onnx format.

Steps:

  1. Create a docker container with the following docker image: nvcr.io/nvidia/pytorch:21.08-py3
  2. In your workspace, clone this repo using git clone https://github.com/nachi-hebbar/yolo-to-onnx
  3. Install dependencies using pip install -r requirements.txt
  4. Run the onnx-converter.sh script
  5. Enter your model information as and when prompted by the onnx_converter script

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages