ODCA-YOLO: An Omni-Dynamic Convolution Coordinate Attention-Based YOLO for Wood Defect Detection

Autor: Rijun Wang, Fulong Liang, Bo Wang, Xiangwei Mou
Jazyk: angličtina
Rok vydání: 2023
Předmět:
Zdroj: Forests, Vol 14, Iss 9, p 1885 (2023)
Druh dokumentu: article
ISSN: 1999-4907
DOI: 10.3390/f14091885
Popis: Accurate detection of wood defects plays a crucial role in optimizing wood utilization, minimizing corporate expenses, and safeguarding precious forest resources. To achieve precise identification of surface defects in wood, we present a novel approach called the Omni-dynamic convolution coordinate attention-based YOLO (ODCA-YOLO) model. This model incorporates an Omni-dimensional dynamic convolution-based coordinate attention (ODCA) mechanism, which significantly enhances its ability to detect small target defects and boosts its expressiveness. Furthermore, to reinforce the feature extraction and fusion capabilities of the ODCA-YOLO network, we introduce a highly efficient features extraction network block known as S-HorBlock. By integrating HorBlock into the ShuffleNet network, this design optimizes the overall performance. Our proposed ODCA-YOLO model was rigorously evaluated using an optimized wood surface defect dataset through ablation and comparison experiments. The results demonstrate the effectiveness of our approach, achieving an impressive 78.5% in the mean average precision (mAP) metric and showing a remarkable 9% improvement in mAP compared to the original algorithm. Our proposed model can satisfy the need for accurate detection of wood surface defects.
Databáze: Directory of Open Access Journals