Laser & Optoelectronics Progress, Volume. 61, Issue 18, 1812006(2024)

3D Object Detection Algorithm Based on Improved YOLOv5

Xueqing Sheng, Shaobin Li*, Jinyan Qu, and Liu Liu
Author Affiliations
  • School of Electronic and Information Engineering, Beijing Jiaotong University, Beijing 100044, China
  • show less

    To address the challenge of handling large volumes of point cloud data for three-dimensional (3D) object detection and the limited effectiveness in detecting small objects, in this study, an enhanced 3D target detection method is proposed that improves the YOLOv5 network based on the idea of Complex-YOLO algorithm. The proposed approach first tackles the issue of lengthy processing times due to extensive point cloud data by adopting the Complex-YOLO strategy of converting point cloud data into an RGB-Map format, which is more manageable for the YOLOv5 network. Enhancements to YOLOv5 include an angle prediction branch and a rotation frame regression loss function to accurately position rotating targets within the RGB-Map. Additionally, the YOLOv5 architecture is modified to better detect small objects by incorporating a feature fusion layer and a dedicated prediction head, which heightens the network's sensitivity to smaller targets. Furthermore, the convolutional block attention module (CBAM) attention mechanism is integrated into the network's neck to further enhance detection sensitivity. Experimental evaluations on the KITTI dataset confirm the superiority of the modified YOLOv5 method over the original Complex-YOLO, with improvements in mean average precision (mAP): Car type mAP increased by 7.48 percentage points, Pedestrian type by 12.54 percentage points, Cyclist type by 1.2 percentage points, and an overall increase of 7.08 percentage points across all categories, demonstrating the effectiveness of this algorithm.

    Keywords
    Tools

    Get Citation

    Copy Citation Text

    Xueqing Sheng, Shaobin Li, Jinyan Qu, Liu Liu. 3D Object Detection Algorithm Based on Improved YOLOv5[J]. Laser & Optoelectronics Progress, 2024, 61(18): 1812006

    Download Citation

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

    Category: Instrumentation, Measurement and Metrology

    Received: Jan. 5, 2024

    Accepted: Feb. 21, 2024

    Published Online: Sep. 14, 2024

    The Author Email: Shaobin Li (shbli@bjtu.edu.cn)

    DOI:10.3788/LOP240451

    CSTR:32186.14.LOP240451

    Topics