关键词:
目标检测
深度学习
YOLOv5s
神经网络
摘要:
目标检测是计算机视觉的一个重要方面,在准确性和鲁棒性方面取得了重大进展。尽管取得了这些进步,但实际应用仍然面临着显著的挑战,主要是小物体的检测不准确或漏检。此外,检测模型的大量参数计数和计算需求阻碍了它们在资源有限的设备上的部署。本文提出了一种基于YOLOv5的高级目标检测模型YOLO-QCK。我们首先在颈部网络金字塔结构中引入一个额外的小物体检测层,从而产生一个更大尺度的特征图,以识别小物体的更精细的特征。此外,我们将C3CrossCovn模块集成到骨干网中。该模块采用滑动窗口特征提取,有效地减少了计算量和参数数量,使模型更加紧凑。与基线YOLOv5s模型相比,我们新开发的模型YOLO-QCK在MS COCO验证数据集上显示出相当大的改进,mAP@0.5增加了4.6%,mAP@0.5:0.95增加了4%,同时保持模型尺寸紧凑,参数为9.49 M。结果验证了YOLO-QCK模型在小目标检测中的高效性能,以较少的参数和计算量实现了高精度。