Table 5 The ablation study of YOLO models on the validation and the test set.

YOLO Model	Data	Validation set (mAP)				Test set (mAP)
YOLO Model	Data	Pink	Purple	Orange	Average	Pink	Purple	Orange	Average
YOLOv4	Aug	0.662	0.657	0.55	0.623	0.527	0.225	0.471	0.408
	Aug + Gaussian	0.927	0.687	0.607	0.74	0.753	0.271	0.316	0.447
	Aug + Normalize	0.877	0.687	0.573	0.712	0.694	0.414	0.297	0.468
	IEP-CDS	0.947	0.723	0.634	0.768	0.742	0.386	0.336	0.488
YOLOv7	Aug	0.837	0.67	0.622	0.71	0.02	0.03	0.013	0.021
	Aug + Gaussian	0.939	0.778	0.572	0.763	0.73	0.427	0.375	0.51
	Aug + Normalize	0.934	0.735	0.622	0.763	0.757	0.435	0.357	0.516
	IEP-CDS	0.918	0.828	0.575	0.774	0.728	0.359	0.397	0.495
YOLOv8	Aug	0.899	0.674	0.612	0.728	0.28	0.403	0.444	0.376
	Aug + Gaussian	0.848	0.663	0.572	0.694	0.683	0.149	0.309	0.38
	Aug + Normalize	0.899	0.696	0.528	0.708	0.616	0.288	0.331	0.412
	IEP-CDS	0.902	0.775	0.64	0.772	0.647	0.306	0.375	0.443

Quick links

Search