风格迁移增强的机场目标检测方法研究

STYLE TRANSFER INSPIRED AIRPORT OBJECT DETECTION

  • 摘要: 在基于图像的目标检测中,机场是一类非常重要的目标,对其进行自动识别具有重要意义。针对一般检测算法难以使用复杂的近地航拍图像中边缘信息的问题,提出一种融合风格迁移增强边缘特征提取的目标检测算法。使用生成对抗网络抑制图像噪声,使用边缘检测算法突出图像边缘特征,将突出边缘信息的图像经由目标检测算法完成机场位置检测。在机场目标检测数据集实验中,结合所提出的边缘特征提取方法的目标检测算法相比原始目标检测算法有精度上的提升,其中结合该特征提取方法的YOLOv5算法的平均精度达到97.7%,验证了该特征提取方法对机场目标检测具有很好的效果。

     

    Abstract: In image-based object detection, the airport is an important kind of object, and automatic recognition of it is of great significance. Aimed at the difficulty of general detection algorithms to correctly extract edge information from complex aerial images, a new object detection algorithm based on style transfer is proposed to enhanced edge feature extraction. The image noise was suppressed by using the generative adversarial network. The noise-suppressed image was transported to an edge detection algorithm to highlight the edge features. The highlighted images were used to complete the airport location detection through the object detection algorithm. In the airport object detection experiment, the object detection algorithm combined with the edge feature extraction approach proposed in this paper has higher precision than the original object detection algorithm. The average precision of YOLOv5 and the proposed feature extraction fusion algorithm achieves 97.7%. Experimental results demonstrate that this feature extraction approach has a good effect on airport object detection.

     

/

返回文章
返回