Optoelectronics Letters, Volume. 20, Issue 6, 367(2024)

YOLOV5s object detection based on Sim SPPF hybrid pooling

Xiuhuan DONG... Shixin LI* and Jixiang and ZHANG |Show fewer author(s)
Author Affiliations
  • Electronic Engineering Department, Tianjin University of Technology and Education, Tianjin 300000, China
  • show less

    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.

    Tools

    Get Citation

    Copy Citation Text

    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

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Received: Aug. 24, 2023

    Accepted: Dec. 12, 2023

    Published Online: Aug. 23, 2024

    The Author Email: Shixin LI (0422211032@tute.edu.cn)

    DOI:10.1007/s11801-024-3170-x

    Topics