DWCA-YOLOv5: An Improve Single Shot Detector for Safety Helmet Detection

Table 2

Effect evaluation of different models on the test set.

Detection model	Clustering method	AP50/%		mAP/%
		Hat	Person
Original YOLOv5	-means	93.3	91.7	92.7
Original YOLOv5	-means++	94.4	92.8	93.6
Improved YOLOv5	-means	95.5	94.6	95.1
Improved YOLOv5	-means++	96.5	95.2	95.9