YOLOV5s object detection based on Sim SPPF hybrid pooling
DOI:
Author:
Affiliation:

Tianjin University of Technology Education

Clc Number:

Fund Project:

[2]国家重点研发计划(2022YFF0706000)

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

    Aiming at the problem of low surface defect detection accuracy of industrial products, an object detection method based on Sim SPPF hybrid pooling improved YOLOV5s model is proposed. The algorithm introduces Channel Attention module, Simplified Spatial Pyramid Pooling Fast Feature Vector Pyramid and Efficient Intersection Over Union loss function. Feature vector pyramids fuse high-dimensional and low-dimensional features, which makes semantic information richer. The channel attention mechanism performs maximum pooling and average pooling operations on the feature map. Hybrid pooling comprehensively improves detection computing efficiency and accurate deployment ability. The results show that the improved YOLOV5s model is better than the original YOLOV5s model, the average test accuracy (mAP) can reach 91.8%, the average accuracy (mAP) of the model can be increased by 17.4%, the detection speed can reach 108 FPS, the detection speed can be increased by 18 FPS, and the improved model is practicable and the overall performance is better than other conventional models.

    Reference
    Related
    Cited by
Get Citation
Share
Article Metrics
  • Abstract:
  • PDF:
  • HTML:
  • Cited by:
History
  • Received:August 24,2023
  • Revised:October 16,2023
  • Adopted:November 07,2023
  • Online:
  • Published: