
学术探讨-Attention-YOLO:融合注意力机制的改进型YOLO目标检测算法.pdf
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
本文介绍了一种创新的目标检测算法——Attention-YOLO,该算法在经典YOLO模型基础上引入了注意力机制,有效提升了复杂场景下的目标识别精度和速度。
实时目标检测算法YOLOv3具有较快的检测速度和良好的精度,但存在边界框定位不够精确、难以区分重叠物体等问题。为此提出了Attention-YOLO算法,该算法借鉴了基于项的注意力机制,并将通道注意力及空间注意力机制加入到特征提取网络中。通过使用经过筛选加权后的特征向量替换原有的特征向量进行残差融合,并添加二阶项来减少信息损失和加速模型收敛。
实验结果显示,在COCO和PASCAL VOC数据集上,Attention-YOLO算法有效降低了边界框的定位误差并提升了检测精度。与YOLOv3相比,在COCO测试集中mAP@IoU[0.5:0.95]最高提高了2.5 mAP;在PASCAL VOC 2007测试集上达到了最高的81.9 mAP。
全部评论 (0)
还没有任何评论哟~


