注意力增强YOLOv5

检测精度高且速度达标

面向连续铸钢板坯检测的改进YOLOv5模型,引入VCIoU损失和注意力机制,在自建数据集上mAP@50:95达到87.75%,比YOLOv5s提高1.58%。

Hongliang, Yang · 方一鸣 · 刘乐 · Hao, Ju · Kang, Kesong

IEEE Transactions on Instrumentation and Measurement 2023

参数信息

mAP
87.75 %
YOLOv s
1.58 %

技术优势

精度提高

引入VCIoU损失,利用顶点与中心点距离信息,在不增加计算成本的情况下提高检测精度。

结构增强

加入注意力机制和BiFPN,提升网络对关键特征的提取能力。

速度达标

在提高精度的同时满足连铸现场对检测速度的要求。

应用场景