YOLOv5改进--轻量化YOLOv5s模型

作者：菜鸟追梦旅行 | 2024-04-08 17:06:37

踩

1、前言

在边缘设备的场景下，目前的YOLOv5s，虽然能够快速实现目标检测，但是运行速度依旧稍慢点，本文在牺牲一点精度前提下，提高目标检测速度，即轻量化YOLOv5s模型，并部署到边缘设备上，可以在CPU上达到实时的检测效果，满足业务的性能需求。

轻量化思路：

1、改进锚框，将对应的锚框全部减半

2、将yolov5s的模型的channels通道数全部都减少一半。

3、训练时，输入图片大小为320，即从 640x640 变为 320x320

原始yolov5s.yaml模型结构

# YOLOv5 声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/菜鸟追梦旅行/article/detail/387005

推荐阅读

相关标签