基于改进YOLOv5s的目标检测算法.docxVIP

机械工程师ME

机械工程师

MECHANICALENGINEER

机械工程师MECHANICAL

机械工程师

MECHANICALENGINEER

PAGE64圆园

PAGE64

圆园24年第9期网址：电邮：hrbengineer@163.com

PAGE65网址：

PAGE65

网址：电邮：hrbengineer@163.com圆园24年第9期

基于改进YOLOv5s的目标检测算法

宋为，王赫莹，郭忠峰，胡克腾

（沈阳工业大学机械工程学院，沈阳110870）

摘要：针对YOLOv5s算法定位能力不足等问题，在YOLOv5s的基础上提出一种改进算法。该算法首先使用CIOU_Loss替换YOLOv5的GIOU_Loss作为算法的边界回归损失函数，考虑到预测框与真实边界框的长宽比与中心距离，可以在提升定位精度的同时提高算法的收敛速度；最后使用轻量化卷积模块GhostConv替换YOLOv5s特征融合层的卷积，在轻量化的同时，可以提升检测精度。使用VOC2007数据集来验证算法的有效性，经试验表明，改进后的算法精度达到了95.8%，比原算法提高了2.3%。

关键词：目标检测；YOLOv5s；损失函数；GhostConv

中图分类号:TP391.4 文献标志码:粤 文章编号：员园园圆原圆猿猿猿（圆园24）09原园