Application of YOLOv3 in Safety Helmet Wearing Detection
CSTR:
Author:
Affiliation:

Clc Number:

TP29

Fund Project:

  • Article
  • |
  • Figures
  • |
  • Metrics
  • |
  • Reference
  • |
  • Related
  • |
  • Cited by
  • |
  • Materials
  • |
  • Comments
    Abstract:

    In view of the complex construction site environment and the problems of the YOLOv3 helmet wearing detection algorithm such as low accuracy and poor robustness, an improved YOLOv3 helmet wearing detection algorithm was proposed. The K-means algorithm was used to cluster out anchor boxes, so as to improve the network output scales. The residual block was introduced at the output to form a residual module; at the same time, the classification loss function was improved to balance the influence of positive and negative samples and difficult and easy samples on the model. In order to verify the effectiveness of the method, it was verified on the NVIDIA GTX1660Ti. The experimental results show that the mAP of the improved YOLOv3 helmet wearing detection algorithm increase by 4.84%, the detection ability of occluded targets and small targets is improved, and the model has strong robustness.

    Reference
    Related
    Cited by
Get Citation

MA Xiaolu, WANG Mingming, WANG Bing. Application of YOLOv3 in Safety Helmet Wearing Detection[J].,2020,37(4):78-86

Copy
Related Videos

Share
Article Metrics
  • Abstract:
  • PDF:
  • HTML:
  • Cited by:
History
  • Received:August 11,2020
  • Revised:
  • Adopted:
  • Online: December 30,2020
  • Published:
Article QR Code