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

Table 4

Dataset category allocation.

Target category	Training set a target number	A test set the target number	Total number of labeled targets
Wearing helmets category	81836	11316	93152
Not wearing helmets category	98187	12021	110208