Skip to content

hsulin0806/onnx_runtime

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

7 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

ONNX runtime with object detection

Run detections on EfficientDet and YOLOv3 models with onnx. ONNX Tutorials
EfficientDet implementation based from: https://github.com/murdockhou/Yet-Another-EfficientDet-Pytorch-Convert-ONNX-TVM
YOLOv3 implementation based from: https://github.com/AlexeyAB/darknet

Installation

Use the package manager pip.

pip install requirements.txt

Setup

.onnx model

/onnx_runtime/models/myOnnxModel.onnx

test images

/onnx_runtime/data/images

Detection

python detect.py <original framework> <path to onnx model> <path to test images>

Output:

Inference time:  0.18  sec
3 objects identified in source image.

alt text

Credit

Credit to the tutorials and documents provided by onnx.
https://github.com/onnx

About

Easily run inferences on EfficientDet & YOLOv3 models using ONNX

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages