Skip to content

Latest commit

 

History

History
32 lines (27 loc) · 899 Bytes

README.md

File metadata and controls

32 lines (27 loc) · 899 Bytes

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