Skip to content

EscaticZheng/yolov3-tiny-onnx-416

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

9 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

yolov3-tiny-onnx-416

convert darknet to onnx with fixed pads. It means conv and maxpool with no auto_pads but fixed pads for model deploy.

environment

onnx >= 1.9.0 (opset_version >=11, default is 12)
onnxruntime

Usage

  1. download model weights from https://pjreddie.com/media/files/yolov3-tiny.weights.
  2. put yolo_to_onnx.py, weights, cfg in the same directory.
  3. run $python yolo_to_onnx.py --model=yolov3-tiny-416 to generate yolov3-tiny-416.onnx
  4. if you want to test it, run $python img_predict.py

image

Reference

  1. https://github.com/jkjung-avt/tensorrt_demos/tree/master/yolo
  2. https://github.com/htshinichi/onnx-yolov3

About

convert darknet to onnx with fixed pads.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages