Research and training of helmet recognition model based on deep learning
Yingwen Xie · 2024
For the detection of hard hats, the human key point detection model and YOLO3 algorithm based on deep learning are adopted. In this paper, we will introduce a series of work required for the design model of helmet recognition, including the construction of model training environment, data set preparation, model training, model visualization operation and model recognition results. The identification model of safety hat in electric power construction site will be developed based on PyTorch. The following framework depicts the structure of a deep learning model framework based on PyTorch.