地铁场景下基于高分辨率网络的旋转小目标检测

ROTATING SMALL TARGET DETECTION BASED ON HIGH-RESOLUTION NETWORK IN SUBWAY SCENE

  • 摘要: 针对地铁场景下旋转小目标难以检测的问题,提出地铁场景下基于高分辨率网络和旋转目标框的旋转小目标检测方法。采用高分辨率网络和特征金字塔网络特征融合头部以增强特征提取能力,并提出旋转高斯核和旋转目标检测框的策略以适应任何方向的目标检测。实验结果表明这些改进使得模型检测准确率在旋转小目标、倾斜的细长目标检测上远远高于应用水平检测目标框的模型。该方法在地铁场景下能准确地检测水平目标的同时也能准确检测旋转小目标以及细长倾斜目标。

     

    Abstract: Aiming at the problem that rotating small objects are difficult to detect in subway scenes, this paper proposes a rotating small object detection method based on high-resolution networks and rotating target frames in subway scenes. The high-resolution network and FPN feature fusion head were used to enhance the feature extraction ability, and the strategy of rotating Gaussian kernel and rotating target detection frame was proposed to adapt to target detection in any direction. The experimental results show that these improvements make the model detection accuracy in the detection of rotating small targets and oblique slender targets much higher than that of applying horizontal detection target boxes. The method can accurately detect horizontal targets in the subway scene, but also can accurately detect rotating small targets and slender inclined targets.

     

/

返回文章
返回