Onnx - Deposit : Conversion & Inference scripts

😇 Motivation

A single repo with code & some theory to quickly refer when working with Onnx.

🔑 Models

1. Classification
• ResNet - PyTorch to Onnx code
• ResNet - Onnx inference script
2. Object Detection
• YoloV5 - Onnx inference script
3. Semantic Segmentation
• UNet - Tensorflow to Onnx
• UNet - Onnx inference script

💫 How to use?

1. Clone the repo
2. Install requirements.txt (Only for inference)
3. Checkout notes on the top of the scripts before running the codes