-
题名基于YOLO优化的轻量级目标检测网络
被引量：16
- 1
-
-
作者
许虞俊
李晨
-
机构
东南大学电子科学与工程学院
-
出处
《计算机科学》
CSCD
北大核心
2021年第S02期265-269,共5页
-
文摘
目标检测是计算机视觉领域中一个相当活跃的研究领域,通过设计大型的深度卷积神经网络来提高目标检测的精度是一种十分有效的方法,然而目前在内存受限的应用场景中并不支持部署大型目标检测网。针对以上问题,文中提出了一种基于You Only Look Once(YOLO)系列单镜头目标检测网络设计原则的轻量级目标检测网,融合了GhostNet中的Ghost Module模块,并参考了MobileNet-v3中的通道注意力模块SE(Squeeze-and-Excitation),在卷积块中加入更优的ECA(Efficient Channel Attention)模块可以更好地利用可用的网络容量,使得网络在减少体系结构和计算的复杂度以及提升模型性能之间实现强的平衡;并且采用了Distance-IoU loss来解决检测框定位不准的问题,有效地提升了网络的收敛速度。最终模型的参数数量被压缩到了1.54 MB,小于YOLO Nano(即4.0MB),并且在VOC2007测试集上的mAP达到了72.1%,高于现有的YOLO Nano(即69.1%)。
-
关键词
目标检测
轻量级
YOLO深度卷积神经网络
Pascal
VOC
-
Keywords
Object detection
Light-weight
YOLO deep convolutional neural network
Pascal VOC
-
分类号
TP391
[自动化与计算机技术—计算机应用技术]
-
