基于特征重构的YOLO目标检测模型实现

Implementation of YOLO Target Detection Model Based on Feature Reconstruction

摘要: 目标检测是计算机视觉领域的四大核心任务之一，它涵盖了目标的分类和定位，至今已有近二十年的研究历史。YOLOv5在YOLO系列目标检测算法中广泛应用，这得益于其稳定的工程实践能力，以及可以较好地平衡检测精度和速度，但YOLOv5算法的检测精度距两阶段目标检测器的性能还有差距。因此本文选取其作为研究对象，对其进行改进，力求在保证检测速度基本不变的同时尽可能地提升精度。针对YOLOv5s目标检测算法不能充分利用特征信息的问题，对YOLOv5s进行改进后，提出一种基于特征重构模块的目标检测模型(F-YOLOv5s)。该特征重构模块将w-h平面上的特征信息转移到空间维度，能够减少下采样带来的信息损失，从而提高目标检测的准确率。实验表明，在PASCAL VOC2007和VOC2012数据集上，本文提出的特征重构模块能有效提高特征信息的利用率，使得检测精度大幅度提升。

Abstract: Object detection is one of the four core tasks in the field of computer vision, which covers the classification and localization of objects, and has been studied for nearly two decades. YOLOv5 is widely used in the YOLO series target detection algorithms, which is due to its stable engineering practice ability and good balance between detection accuracy and speed. However, the detection accuracy of YOLOv5 algorithm is still short of the performance of the two-stage target detector. Aiming at the problem that the YOLOv5 target detection algorithm cannot make full use of feature information, an improved YOLOv5 target detection model based on feature reconstruction module (F-YOLOv5) is proposed. The feature reconstruction module transfers feature information from the w-h plane to the spatial dimension, reducing information loss caused by down sampling and thereby improving the accuracy of object detection. Experiments show that on PASCAL VOC2007 and VOC2012 data sets, the feature reconstruction module proposed in this paper can effectively improve the utilization rate of feature information and greatly improve the detection accuracy.