Skip to content

VITA-Alchemy/yolov5_6.0_opencvdnn_python

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

15 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

yolov5_6.0_opencvdnn_python

基于numpy改写ultralytics/yolov5 v6.0 opencv推理代码,无需依赖pytorch;前处理后处理使用numpy代替了pytroch;故无需修改网络结构或代码,直接使用。

目前适用于:v6.0 v6.1 版本

Environment

pytorch >= 1.7.0    # export.py  need
opencv-python >= 4.5.4
onnxruntime == 1.5.2

RUN

git clone -b v6.0 https://github.com/ultralytics/yolov5.git
download https://github.com/ultralytics/yolov5/releases/download/v6.0/yolov5s.pt
python export.py --simplify --opset 12 
#官方版测试
python detect.py --weights ./yolov5s.onnx --dnn
python main_dnn.py

Reference

https://github.com/ultralytics/yolov5
https://github.com/hpc203/yolov5-dnn-cpp-python

About

yolov5-v6.0 opencv dnn

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages