当前位置:   article > 正文

分享|YOLOv5发布第六个版本,支持一键适配OpenVINO/OpenCV DNN部署_yolov5n

yolov5n

新特性

首次发布支持移动端的最小化模型YOLOv5n,只有1.9MB,全称为YOLOv5 Nano。支持模型导出为tensorflow跟keras格式,这是什么鬼,就这招彻底打废原来的tensorflow跟keras版本模仿者!OpenCV DNN支持,原来导出ONNX要支持OpenCV DNN比较麻烦,需要开发者自己一通猛改,现在容易多了,支持OpenCV DNN与ONNXRUNTIME。总结一下主要有三点:

更小型化的模型YOLOv5n,移动端治愈

支持导出tensorflow/keras格式模型文件

导出onnx支持opencv dnn,再也不用开发者自己修改了!

还有更惊喜的,模型更加轻量化、精度更高、速度更快(都是跟之前的自己比)
在这里插入图片描述
在这里插入图片描述
模型与源码下载,一键直达:

https://github.com/ultralytics/yolov5/releases/tag/v6.0
测试演示
  • 1
  • 2

测试演示

下载源码之后,一张自带测试图象,

python detect.py --weights yolov5n.pt --source data/images/zidane.jpg --img 640
  • 1

感觉还很好!
在这里插入图片描述
视频测试:
在这里插入图片描述
导出为ONNX格式并测试

1.python export.py --weights yolov5n.pt --img 640 --batch 1
2.export failure: Unsupported ONNX opset version: 13
  • 1
  • 2

导出失败了,原因是pytorch中onnx opset版本太高了,不支持!还好有个命令行参数可以指定导出版本

python detect.py --weights yolov5n.onnx --dnn --source D:/images/city-walk.png --img 640
  • 1

在这里插入图片描述
测试视频文件:


python detect.py --weights yolov5s.onnx --dnn --source D:/images/video/SungEun.avi
  • 1
  • 2

运行结果:
在这里插入图片描述

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/繁依Fanyi0/article/detail/147459
推荐阅读
相关标签
  

闽ICP备14008679号