摘要
小目标检测一直是目标检测领域的难点,针对卷烟厂卷包车间摄像头安装位置较高、小目标检测精度低和总体检测精度较低的问题,提出了一种改进的YOLOv8n目标检测算法YOLOv8n-FIAL。首先使用添加通道重排机制的C2fg模块代替原本C2f模块,提高特征学习能力,使用自适应通道特征融合模块代替YOLOv8n算法Neck部分的Concate操作,使特征融合更加充分;然后增加小目标检测层,提高小目标检测精度,降低漏检率;最后使用Focal-EIOU损失函数替换原来的CIOU损失函数,平衡锚框与真实框重叠较大的高质量锚框的数量远少于低质量锚框训练实例不平衡的问题。实验结果表明,在自制的卷烟厂工人违规作业数据集上,所提出的YOLOv8n-FIAL检测方法相比原始的YOLOv8n方法的总体平均精度均值提升了7.6%,对口鼻、手拿手机和衣服领口这3类小目标平均精度均值提升最大,分别提升了8.3%,8%和9.6%;在公共数据集VOC2007上,YOLOv8n-FIAL算法相比YOLOv8n算法的总体平均精度均值提升了1.6%。
Small object detection has always been a difficult point in the field of object detection.In response to the high installation of cameras in cigarette factory packaging rooms,low accuracy of small object detection,and overall low detection accuracy,an improved YOLOv8n object detection algorithm YOLOv8n FIAL has been proposed.Firstly,the C2fg module with added channel rearrangement mechanism is used to replace the original C2f module to improve feature learning ability.The adaptive channel feature fusion module is used to replace the Concate operation in the Neck section of the YOLOv8n algorithm,making feature fusion more comprehensive;then,add a small target detection layer to improve the accuracy of small target detection and reduce the missed detection rate;finally,the Focal EIOU loss function is used to replace the original CIOU loss function.The number of high-quality anchor boxes with a large overlap between the balanced anchor box and the real box is much less than the problem of imbalanced training instances of low-quality anchor boxes.The experimental results show that on the self-made cigarette factory worker violation operation dataset,the YOLOv8n FIAL detection method proposed in this article has an overall average accuracy improvement of 7.6%compared to the original YOLOv8n method.The average accuracy improvement for the three types of small targets,namely mouth,nose,handheld phone,and clothing collar,is the largest,with increases of 8.3%,8%,and 9.6%,respectively;On the public dataset VOC2007,the YOLOv8n FIAL algorithm has an overall average accuracy improvement of 1.6%compared to the YOLOv8n algorithm.
作者
刘恒
林虹宇
吴涛
LIU Heng;LIN Hongyu;WU Tao(School of Electrical Information,Southwest Petroleum University,Chengdu 610500,China)
出处
《计算机科学》
CSCD
北大核心
2024年第S01期541-548,共8页
Computer Science