计算机科学
目标检测
特征提取
人工智能
计算复杂性理论
模式识别(心理学)
特征(语言学)
卷积(计算机科学)
小波
计算机视觉
代表(政治)
小波变换
领域(数学)
分割
离散小波变换
编码(集合论)
源代码
对象(语法)
计算
卷积神经网络
棱锥(几何)
入侵检测系统
作者
Dan Wu,Wenhao Wu,Haoye Zheng,Fei Wang,Guowang Gao
摘要
ABSTRACT The detection of small objects in remote sensing imagery presents significant challenges, mainly attributed to intricate background complexities, inadequate feature extraction performance, and constrained receptive field dimensions. Moreover, under constrained computational resources, improving detection accuracy while reducing the model's computational complexity and parameter count has become an urgent issue. In response to these challenges, a novel lightweight model specifically designed for small object detection, named DWT‐YOLO, is proposed in this study. By introducing the Discrete Wavelet Transform (DWT) combined with convolutional down‐sampling operations, the model effectively expands the receptive field and enhances feature extraction capabilities. Simultaneously, an innovative Wavelet Convolution Fusion module (WCFM) is designed, which improves multi‐scale feature fusion and cross‐domain information association, enhancing the feature representation for small objects while keeping model complexity relatively low. Additionally, the model incorporates a Spatial‐Channel Decoupling Down‐sampling (SCDown) module, which effectively reduces redundant parameters and improves detection accuracy. We first validated the effectiveness of DWT‐YOLO on the RSOD dataset and then further assessed its small object detection performance on the VisDrone2019 dataset. The proposed model demonstrates a 5.4% enhancement in detection accuracy compared to the baseline model. The model has only 5.8M parameters, and it outperforms several advanced models in detection performance. Experimental results demonstrate that DWT‐YOLO not only achieves significant improvements in small object detection tasks but also exhibits lightweight characteristics, providing new solutions and technical insights for this field. The implementation source code of the proposed method is publicly accessible at the following repository: http://github.com/zzgithubFly/DWT‐YOLO .
科研通智能强力驱动
Strongly Powered by AbleSci AI