复杂光照场景下的自动驾驶汽车行人检测

PEDESTRIAN DETECTION IN SELF-DRIVING CARS IN COMPLEX LIGHTING SCENES

  • 摘要: 针对在低光照或光照条件变化剧烈的场景下行人检测易出现漏检、定位不准问题,提出基于图像增强(EnlightenGAN)与YOLOv5联合学习的行人检测方法,该方法中的特征提取网络可以学习到经过对抗神经网络增强模块重构的图像结构细节和颜色特征。采用平滑处理后的相对熵损失函数作为YOLOv5的置信度损失,以提升网络的泛化性能。实验结果表明,基于图像增强与YOLOv5联合学习的方法在自制的复杂光照场景行人数据集上有效降低不同光照对检测行人的敏感性,精确率达到87.58%;在Caltech数据集上改进YOLOv5损失后收敛加快且检测精确率提升1.73百分点。

     

    Abstract: For pedestrian detection in low light or scenes with drastic changes in light conditions, the problems of missed detection and inaccurate positioning are likely to occur. A pedestrian detection method based on the joint learning of image enhancement (EnlightenGAN) and YOLOv5 is proposed. The feature extraction network in this method could learn the structural details and color features of the image reconstructed by the enhancement module of the adversarial neural network. The relative entropy loss function after smoothing was used as the confidence loss of YOLOv5 to improve the generalization performance of the network. The experimental results show that the method based on image enhancement and YOLOv5 joint learning can effectively reduce the sensitivity of different illumination to pedestrian detection on the self-made complex illumination scene pedestrian data set, and the precision reaches 87.58%. After improving the YOLOv5 loss on the Caltech data set, the convergence speeds up and the detection precision is increased by 1.73 percentage points.

     

/

返回文章
返回