Research Article
Automatic Fabric Defect Detection Based on an Improved YOLOv5
Table 1
Ablation study of the teacher network on the Tianchi AI database.
| Attention module | Multitask learning | Focal loss | Central loss constraints | AUC | mAP |
| | | | | 0.938 | 0.403 | √ | | | | 0.957 | 0.412 | √ | √ | | | 0.965 | 0.431 | √ | √ | √ | | 0.971 | 0.441 | √ | √ | | √ | 0.973 | 0.442 | √ | √ | √ | √ | 0.981 | 0.447 |
|
|