CBAM-YOLOv5: A Promising Network Model for Wear Particle Recognition

Table 2

Results of ablation experiments.

Model	Baseline network	AHE	CBAM	DWConv	DIOU-LOSS			mAP		FPS
1	YOLOV5					0.886	0.843	0.8376	0.029	34.87
2	YOLOV5	√				0.902	0.858	0.8468	0.029	34.87
3	YOLOV5		√			0.929	0.842	0.8663	0.038	26.32
4	YOLOV5			√		0.894	0.875	0.8675	0.017	58.82
5	YOLOV5		√	√		0.928	0.839	0.8673	0.023	43.48
6	YOLOV5				√	0.910	0.844	0.8415	0.034	29.41
7	YOLOV5	√	√	√	√	0.967	0.916	0.9262	0.031	32.26