Wood defect detection plays a critical role in ensuring wood quality and minimizing resource waste in industrial production. Advanced object detectors show excellent performance in wood defect detection. However, the texture of wood defects is complex and categories are diverse, and existing methods still face challenges in the detection of defects with small size and unclear boundaries. To address the issue, we propose a novel wavelet detector based on DETR for real-time wood defect detection, which is called WT-RT-DETR. Firstly, we optimize the network structure by incorporating Partial Self Attention (PSA) as the object detection head, which enhances the multi-scale shallow and deep features. Secondly, we propose an Inverse Haar Wavelet Transform Upsampling module (UP-IHWT), which preserves details and multi-scale information based on the wavelet inverse transform. Finally, we propose the Reparameterized Wavelet Transform Convolution 3 module (RepWTC3) by employing wavelet convolution into RepC3, which strengthens the boundary features extraction capabilities of the neck network. The experimental results show that the mAP50 of WT-RT-DETR is 69.1%, which is higher than the state-of-the-art methods, and the model parameters are slightly reduced. It can better balance efficiency and accuracy.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

WT-RT-DETR: A Wavelet Detector for Real-Time Wood Defect Detection

  • Yuefeng Zhao,
  • Junjie Wang,
  • Jinwei Zhang,
  • Qifei Wang,
  • Xianqi Meng,
  • Guicong Zhang,
  • Yuxin Song,
  • Nannan Hu

摘要

Wood defect detection plays a critical role in ensuring wood quality and minimizing resource waste in industrial production. Advanced object detectors show excellent performance in wood defect detection. However, the texture of wood defects is complex and categories are diverse, and existing methods still face challenges in the detection of defects with small size and unclear boundaries. To address the issue, we propose a novel wavelet detector based on DETR for real-time wood defect detection, which is called WT-RT-DETR. Firstly, we optimize the network structure by incorporating Partial Self Attention (PSA) as the object detection head, which enhances the multi-scale shallow and deep features. Secondly, we propose an Inverse Haar Wavelet Transform Upsampling module (UP-IHWT), which preserves details and multi-scale information based on the wavelet inverse transform. Finally, we propose the Reparameterized Wavelet Transform Convolution 3 module (RepWTC3) by employing wavelet convolution into RepC3, which strengthens the boundary features extraction capabilities of the neck network. The experimental results show that the mAP50 of WT-RT-DETR is 69.1%, which is higher than the state-of-the-art methods, and the model parameters are slightly reduced. It can better balance efficiency and accuracy.