Research Article

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

Table 5

Experiment operating environment.

CategoryEntryVersion

Hardware configurationSystemUbuntu 18.04
GPUGeForce RTX 2080 Ti
Software configurationCPUAMD Ryzen 7 3800X 8-Core
Python version3.8
Deep learning frameworkPytorch
CUDA10.0