CTDD-YOLO: A Lightweight Detection Algorithm for Tiny Defects on Tile Surfaces
To address the challenge of detecting tiny flaws in tile defect detection, a lightweight algorithm for identifying minor defects in tile images has been developed, referred to as CTDD-YOLO. Firstly, CAACSPELAN is proposed as the core component of the backbone network for extracting features of tile...
Gespeichert in:
Veröffentlicht in: | Electronics (Basel) 2024-10, Vol.13 (19), p.3931 |
---|---|
Hauptverfasser: | , , , |
Format: | Artikel |
Sprache: | eng |
Schlagworte: | |
Online-Zugang: | Volltext |
Tags: |
Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
|
Zusammenfassung: | To address the challenge of detecting tiny flaws in tile defect detection, a lightweight algorithm for identifying minor defects in tile images has been developed, referred to as CTDD-YOLO. Firstly, CAACSPELAN is proposed as the core component of the backbone network for extracting features of tile defects; secondly, full-dimensional dynamic convolution ODConv is introduced at the end of the backbone network to enhance the model’s ability to deal with tiny defects; next, a new neck network, CGRFPN, is proposed to improve the model’s ability to represent multi-scale features and enhance the model’s ability to recognize small targets in the context of large formats; finally, MPNWD is proposed to optimize the loss function to improve the model’s detection accuracy further. Experiments on the Ali Tianchi tile defect detection dataset show that the CTDD-YOLO model not only has a lower number of parameters than the original YOLOv8n but also improves the mAP by 7.2 percentage points, i.e., the proposed model can more accurately recognize and deal with minor surface defects of tiles and can significantly improve the detection effect while maintaining the light weight. |
---|---|
ISSN: | 2079-9292 2079-9292 |
DOI: | 10.3390/electronics13193931 |