AW-YOLO:一种结构引导与频域增强的陨石坑检测网络

AW-YOLO: a crater detection network with structure guidance and frequency domain enhancement

  • 摘要: 高精度、高效的陨石坑自动检测对于行星地质研究、地表年龄估算及行星车自主导航至关重要. 然而,现有基于深度学习的检测方法在应对行星遥感影像中陨石坑的边缘模糊、尺度差异巨大等挑战时,往往存在特征表达不充分的问题. 针对上述问题,本文提出一种面向火星热红外遥感图像的陨石坑检测方法. 针对陨石坑边缘模糊及结构退化问题,设计了一种Sobel边缘引导注意力(Sobel edge-guided attention, SEGA)模块. 在骨干网络中引入基于Sobel算子的环形边缘先验信息,并结合交叉注意力机制,引导网络聚焦于陨石坑的轮廓结构特征. 针对陨石坑尺度跨度大及特征表达不足的问题,构建了基于小波变换的特征融合模块(wavelet fusion module,WFM),通过引入小波变换提取的多分辨率频域信息与通道注意力机制结合,增强对陨石坑特征的融合能力,并有效应对从微型坑到大型撞击坑的极端尺度变化. 在火星日陨石坑检测数据集上的实验结果表明,与原始YOLO系列模型及其他主流检测模型相比,本文所提出的AW-YOLO(you only look once with attention and wavelet)模型在保持较高推理效率的同时,取得较好性能,其中准确率(Precision)、召回率(Recall)和F1分数分别达到0.88、0.87和0.87.

     

    Abstract: High-precision and efficient automatic crater detection is crucial for planetary geological research, surface age estimation, and rover autonomous navigation. However, deep learning-based detection methods often suffer from insufficient feature representation when addressing challenges such as blurred crater edges and huge scale variations in planetary remote sensing images. To tackle these issues, this paper proposes a crater detection method for Martian thermal infrared remote sensing images. A Sobel edge-guided attention (SEGA) module is designed to alleviate the problems of blurred edges and degraded structures of craters. By introducing ring-shaped edge prior information based on the Sobel operator into the backbone network and combining it with a cross-attention mechanism, the network is guided to focus on the contour and structural features of craters. To address the large-scale span of craters and insufficient feature expression, a wavelet feature fusion module (WFM) is constructed. By integrating multi-resolution frequency-domain information extracted via wavelet transform with a channel attention mechanism, the module enhances the feature fusion capability for craters and effectively copes with extreme scale variations from mini-craters to large impact craters. Experimental results on the public Mars Day Crater Detection dataset demonstrate that, compared with the original YOLO series models and other mainstream detection models, the proposed AW-YOLO (you only look once with attention and wavelet) model achieves favorable performance while maintaining high inference efficiency. Specifically, its Precision, Recall and F1-score reach 0.88, 0.87 and 0.87, respectively.

     

/

返回文章
返回