Small object detection has important application value in the fields of environmental monitoring, resource detection and analysis, etc. However, the current general object detectors are not very ideal for the detection of small objects. To this end, this paper proposes an efficient, low-complexity, anchor-free small object detection framework YOLO-SOD based on YOLOv8. First, a content-aware feature recombination upsampling operator (CARAFE) is integrated in the upsampling operation part of the framework to achieve more accurate and efficient feature reconstruction. Then, the spatial and channel reconstruction convolutional block (SCConv) is widely integrated in BackBone and Neck to reduce the computational cost caused by redundant feature extraction in visual tasks. Finally, the occlusion-aware attention module (SEAM) is introduced at the end of the detection framework to help the model more accurately identify occluded objects. The ablation experiment on the general small object detection dataset VisDrone2021 proves the effectiveness of several modules introduced in this paper for small object detection. On the VisDrone2021 dataset, YOLO-SOD can achieve an accuracy of 31.1% AP50:95 and 52.3% AP50, which are 2.5% and 3.1% higher than the baseline model respectively.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

YOLO-SOD: Improved YOLO Small Object Detection

  • Yang Xu,
  • Jun Lu,
  • Chao Wang

摘要

Small object detection has important application value in the fields of environmental monitoring, resource detection and analysis, etc. However, the current general object detectors are not very ideal for the detection of small objects. To this end, this paper proposes an efficient, low-complexity, anchor-free small object detection framework YOLO-SOD based on YOLOv8. First, a content-aware feature recombination upsampling operator (CARAFE) is integrated in the upsampling operation part of the framework to achieve more accurate and efficient feature reconstruction. Then, the spatial and channel reconstruction convolutional block (SCConv) is widely integrated in BackBone and Neck to reduce the computational cost caused by redundant feature extraction in visual tasks. Finally, the occlusion-aware attention module (SEAM) is introduced at the end of the detection framework to help the model more accurately identify occluded objects. The ablation experiment on the general small object detection dataset VisDrone2021 proves the effectiveness of several modules introduced in this paper for small object detection. On the VisDrone2021 dataset, YOLO-SOD can achieve an accuracy of 31.1% AP50:95 and 52.3% AP50, which are 2.5% and 3.1% higher than the baseline model respectively.