Research on the Charging Socket Detection Algorithm based on Improved YOLOv5

被引:0
作者
Chen, Guangmeng [1 ]
Lu Sun [1 ]
Wang, Muchen [2 ]
Wang, Zheng [3 ]
机构
[1] Xidian Univ, Xian, Peoples R China
[2] Shanghai Expt Sch, Shanghai, Peoples R China
[3] Hangzhou Meteron Technol, Hangzhou, Peoples R China
来源
2023 28TH ASIA PACIFIC CONFERENCE ON COMMUNICATIONS, APCC 2023 | 2023年
关键词
mobile charging robot; YOLOv5; attention mechanism; loss function;
D O I
10.1109/APCC60132.2023.10460685
中图分类号
TN [电子技术、通信技术];
学科分类号
0809 ;
摘要
With the rapid increase in the number of electric vehicles, mobile charging robots have been attracting widespread attention worldwide. The primary task of a mobile charging robot, which enables intelligent charging for electric vehicles, is to identify and locate the charging ports. Currently, the detection of charging ports relies mainly on traditional algorithms, which suffer from low real-time performance and accuracy. Therefore, this paper proposes an improved algorithm network based on YOLOv5. It incorporates a target detection head (DYHEAD) based on the attention mechanism into YOLOv5s and utilizes SIoU as the loss function to enhance the accuracy and precision of object detection. The experiment shows that the improved algorithm achieved an average precision improvement of 3.4% and an mAP50 improvement of 2.7%. Simultaneously, with the integration of a depth camera, the algorithm successfully recognized and located three different standard charging ports. The recognition frame rate reached 85-90 frames per second, and the detection range exceeded 1.5 meters. Therefore, this algorithm can be used for the detection of charging ports in tasks involving mobile charging robots.
引用
收藏
页码:342 / 346
页数:5
相关论文
共 8 条
  • [1] Single-Stage UAV Detection and Classification with YOLOV5: Mosaic Data Augmentation and PANet
    Dadboud, Fardad
    Patel, Vaibhav
    Mehta, Varun
    Bolic, Miodrag
    Mantegh, Iraj
    [J]. 2021 17TH IEEE INTERNATIONAL CONFERENCE ON ADVANCED VIDEO AND SIGNAL BASED SURVEILLANCE (AVSS 2021), 2021,
  • [2] Dynamic Head: Unifying Object Detection Heads with Attentions
    Dai, Xiyang
    Chen, Yinpeng
    Xiao, Bin
    Chen, Dongdong
    Liu, Mengchen
    Yuan, Lu
    Zhang, Lei
    [J]. 2021 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, CVPR 2021, 2021, : 7369 - 7378
  • [3] Gevorgyan Z, 2022, Arxiv, DOI arXiv:2205.12740
  • [4] Fast R-CNN
    Girshick, Ross
    [J]. 2015 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2015, : 1440 - 1448
  • [5] SSD: Single Shot MultiBox Detector
    Liu, Wei
    Anguelov, Dragomir
    Erhan, Dumitru
    Szegedy, Christian
    Reed, Scott
    Fu, Cheng-Yang
    Berg, Alexander C.
    [J]. COMPUTER VISION - ECCV 2016, PT I, 2016, 9905 : 21 - 37
  • [6] You Only Look Once: Unified, Real-Time Object Detection
    Redmon, Joseph
    Divvala, Santosh
    Girshick, Ross
    Farhadi, Ali
    [J]. 2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, : 779 - 788
  • [7] Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networks
    Ren, Shaoqing
    He, Kaiming
    Girshick, Ross
    Sun, Jian
    [J]. IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2017, 39 (06) : 1137 - 1149
  • [8] Zheng ZH, 2020, AAAI CONF ARTIF INTE, V34, P12993