欢迎访问我的网站,希望内容对您有用,感兴趣的可以加入免费知识星球。

YOLOv5发布v6.1正式版

YOLO 迷途小书童 2年前 (2022-02-22) 3037次浏览 0个评论

环境

  • windows 10 64bit
  • YOLOv5 v6.1

视频看这里

此处是 youtube 的播放链接,需要科学上网。喜欢我的视频,请记得订阅我的频道,打开旁边的小铃铛,点赞并分享,感谢您的支持。

版本更新

就在刚刚,YOLOv5 官方发布了 v6.1 正式版,这是一个小更新,从作者公布的 release note 中,可以看出,网络结构未变,最大的改变就是增加了向其他模型的转换,如 TensorRTEdge TPUOpenVINO,加上之前版本已经支持的模型,目前市面上常见的基本上都已经囊括了。

yolov5 export

模型导出,以 onnx 为例

# 导出onnx,其他模型对应的字段是torchscript,openvino,engine,coreml,saved_model,pb,tflite,edgetpu,tfjs
python export.py --weights yolov5s.pt --include onnx

# 检测
python detect.py --weights yolov5s.onnx

# pytorch hub使用
model = torch.hub.load('ultralytics/yolov5', 'custom', 'yolov5s.onnx')

# 验证
python val.py --weights yolov5s.onnx

# 模型可视化,可以使用之前推荐过的netron,https://netron.app

其他一些更新

  • 提供了转换模型的基准测试, python utils/benchmarks.py --weights yolov5s.pt,目前只支持 CPUGPU要等到下个版本了
  • Hyperparameters小改动,hyp-scratch-large.yaml 中的 lrf 从原来的 0.2 降为 0.1
  • 训练中将 one-cycle cosine 改为 one-cycle linear

更多详尽的版本更新信息请移步 https://github.com/ultralytics/yolov5/releases/tag/v6.1

喜欢 (2)

您必须 登录 才能发表评论!