Journal of Applied Optics, Volume. 45, Issue 5, 975(2024)

Infrared dim target detection method based on improved YOLOv5s model

Jianjun ZHANG1...2, Yudan CHEN1, Yuling LIU3, Mingming ZHANG3 and Fuyu HUANG1,* |Show fewer author(s)
Author Affiliations
  • 1Shijiazhuang Campus, Army Engineering University of PLA, Shijiazhuang 050003, China
  • 2The 54th Research Institute of China Electronics Technology Group Corporation, Shijiazhuang 050081, China
  • 3Beijing Institute of Remote Sensing Equipment, Beijing 100854, China
  • show less
    Figures & Tables(11)
    YOLOv5s model structure
    Improved YOLOv5s model structure
    TCC module structure
    Ghost module network structure
    Improved SPP structure after introducing Ghost module
    Improved CSP2_X structure after introducing Ghost module
    Partial images of dataset
    Comparison of target detection results before and after YOLOv5s improvement
    Comparative experimental results
    • Table 1. Configuration of experimental platform

      View table
      View in Article

      Table 1. Configuration of experimental platform

      名称环境参数
      操作系统Ubuntu
      CPUIntel(R) Xeon(R) Platinum 8255C
      GPUNIVDIA RTX 2080 Ti
      深度学习框架Pytorch1.5.1
      CUDA版本10.1
    • Table 2. Ablation experimental results

      View table
      View in Article

      Table 2. Ablation experimental results

      组号Precision/%Recall/%mAP@0.5/%FPS/(frame·s−1)FLOPS/GParams/M
      YOLOv5s93.5592.2790.047317.3036.627
      YOLOv5s中引入TCC94.9192.9691.917017.0446.303
      YOLOv5s中引入Ghost95.1193.6491.946416.5206.288
      YOLOv5s中引入TCC+Ghost94.8992.6192.306816.2605.964
    Tools

    Get Citation

    Copy Citation Text

    Jianjun ZHANG, Yudan CHEN, Yuling LIU, Mingming ZHANG, Fuyu HUANG. Infrared dim target detection method based on improved YOLOv5s model[J]. Journal of Applied Optics, 2024, 45(5): 975

    Download Citation

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

    Category:

    Received: Mar. 14, 2024

    Accepted: --

    Published Online: Dec. 20, 2024

    The Author Email: HUANG Fuyu (黄富瑜)

    DOI:10.5768/JAO202445.0502005

    Topics