计算机工程与应用 ›› 2022, Vol. 58 ›› Issue (11): 187-192.DOI: 10.3778/j.issn.1002-8331.2012-0521

• 模式识别与人工智能 • 上一篇    下一篇

一种改进复杂场景下小目标检测模型的方法

周慧,严凤龙,褚娜,刘振宇   

  1. 1.大连东软信息学院 软件学院,辽宁 大连 116023 
    2.大连东软信息学院 计算机学院,辽宁 大连 116023
  • 出版日期:2022-06-01 发布日期:2022-06-01

Approach to Improve Detection Model for Small Object in Complex Scenes

ZHOU Hui, YANG Fenglong, CHU Na, LIU Zhenyu   

  1. 1.School of Software, Dalian Neusoft Information University, Dalian, Liaoning 116023, China
    2.School of Computer, Dalian Neusoft Information University, Dalian, Liaoning 116023, China
  • Online:2022-06-01 Published:2022-06-01

摘要: 复杂场景下小目标检测是目标检测领域的研究难点和热点。传统的two-stage和one-stage检测模型都是通过预先设定锚点框与真实目标框的交并比(intersection over union,IoU)阈值来划分正负样本集,同时这组预定义的固定锚点框还用于获取候选框,进而得到检测结果。然而,在复杂场景下,预先设定的IoU阈值会带来正负样本不均衡问题;针对小尺寸目标(船舶)检测,预定义的锚点框也很难保证覆盖目标的位置和密度,因此限制了检测模型的准确率。为了解决上述问题,提出自适应锚点框(adaptive anchor boxes,AAB)的方法优化目标检测网络,采用基于形状相似度距离的聚类算法生成锚点框,提高目标区域定位技术;采用利用聚类的锚点框计算自适应IoU阈值(adaptive threshold selection,ATS),划分正负样本,保证样本均衡。对复杂场景下的小目标(船舶目标)进行检测,实验结果表明,采用自适应锚点框方法和自适应阈值选择方法的目标检测模型在复杂场景中检测均能提升准确,对比faster R-CNN、FPN、Yolo3和pp-Yolo,融合了上述新方法的模型均提升了检测准确率,分别提升了9.6、2.6、9.8和9.9个百分点。

关键词: 小目标检测, 自适应阈值选择, 自适应锚点框

Abstract: Small object detection in complex scenes is a research difficulty and hot spot in the field of object detection. Many current one-stage and two-stage detection models both divide the positive and negative sample sets by pre-setting the intersection over union(IoU) threshold of the anchor box and the real object box. The anchor box is also used to obtain candidate boxes, and then to obtain the detection result. However, in complex scenarios, the preset IoU threshold leads to the imbalance of positive and negative samples. In addition, it is difficult to guarantee the location and density of the covered target with the predefined anchor boxes. To avoid the above problems, this paper proposes an adaptive anchor box(AAB) method to optimize the object detection network, and uses a clustering algorithm based on shape similarity distance to generate anchor boxes to improve object position accuracy. The anchor box of the class calculates the adaptive threshold selection(ATS), and divides the positive and negative samples to ensure sample balance. The experimental results show that the detection of small targets(ship targets) in complex scenes, the object detection model of adaptive anchor frame method and adaptive threshold selection method can improve the accuracy of detection in complex scenes. Compared with faster R-CNN, FPN, Yolo3, and pp-Yolo, the models that incorporate the above new methods have improved the detection accuracy rate by 9.6, 2.6, 9.8 and 9.9?percentage points respectively.

Key words: small object detection, adaptive threshold selection, adaptive anchor boxes