• Optoelectronics Letters
  • Vol. 20, Issue 6, 367 (2024)
Xiuhuan DONG, Shixin LI*, and Jixiang and ZHANG
Author Affiliations
  • Electronic Engineering Department, Tianjin University of Technology and Education, Tianjin 300000, China
  • show less
    DOI: 10.1007/s11801-024-3170-x Cite this Article
    DONG Xiuhuan, LI Shixin, and ZHANG Jixiang. YOLOV5s object detection based on Sim SPPF hybrid pooling[J]. Optoelectronics Letters, 2024, 20(6): 367 Copy Citation Text show less

    Abstract

    Aiming at the problem of low surface defect detection accuracy of industrial products, an object detection method based on simplified spatial pyramid pooling fast (Sim SPPF) hybrid pooling improved you only look once version 5s (YOLOV5s) model is proposed. The algorithm introduces channel attention (CA) module, simplified SPPF feature vector pyramid and efficient intersection over union (EIOU) loss function. Feature vector pyramids fuse high-dimensional and low-dimensional features, which makes semantic information richer. The CA mechanism performs maximum pooling and average pooling operations on the feature map. Hybrid pooling comprehensively improvesdetection 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%, which can be increased by 17.4%, and the detection speed can reach 108 FPS, which can be increased by 18 FPS. The improved model is practicable, and the overall performance is better than other conventional models.
    DONG Xiuhuan, LI Shixin, and ZHANG Jixiang. YOLOV5s object detection based on Sim SPPF hybrid pooling[J]. Optoelectronics Letters, 2024, 20(6): 367
    Download Citation