Helmet wearing detection for cyclists based on improved SSD
DOI:
CSTR:
Author:
Affiliation:

School of Physics & Electronic Information Engineering, Henan Polytechnic University,Jiaozuo 454000,China

Clc Number:

TP391.4

Fund Project:

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

    In order to solve the problems of small target, intensive, poor accuracy, slow detection speed and difficult application in helmet wearing detection task, this paper proposed a EfficientNetV2-SSD algorithm based on SSD network. Aiming at the problem of multiple SSD network parameters, the improved lightweight network EfficientnetV2 is used to replace the feature extraction network in the SSD to reduce network parameters and improve detection speed. For small targets that are difficult to detect, top-down and bottom-up FPN pyramid structures are used to enrich the information of all prediction feature layers to the maximum extent and improve the detection accuracy of small targets. Aiming at the characteristics of the detected target such as helmet, the size and proportion of the prior frame are redesigned to improve the accuracy of small target detection, accelerate the convergence speed of the network and reduce the network volume. The experimental results show that EfficientNetV2-SSD improved the average accuracy of helmet wearing detection by 7.01% and the network volume was reduced by 75% compared with the SSD network, with better practicability.

    Reference
    Related
    Cited by
Get Citation
Share
Article Metrics
  • Abstract:
  • PDF:
  • HTML:
  • Cited by:
History
  • Received:
  • Revised:
  • Adopted:
  • Online: March 19,2024
  • Published:
Article QR Code