一种基于YOLO的隧道障碍物检测方法及系统

本发明属于人工智能技术领域，公开了一种基于YOLO的隧道障碍物检测方法及系统，包括采集图像数据；提取图像的光照特征，获得光照特征图，基于光照特征图统计图像的光照分布，根据图像的光照分布动态调整图像的亮度和对比度，获得第一图像，提取第一图像的特征，获得第一特征图；提取隧道的几何特征，获得隧道位置分布图，将隧道位置分布图与第一特征图进行融合，获得联合特征图，利用联合特征图进行隧道障碍物检测；定义损失函数，优化模型参数，获得训练好的模型；利用训练好的模型对待检测图像进行检测。本发明针对隧道施工场景的特殊性，对YOLOv5模型进行一系列适应性改进，提升了隧道障碍物的检测精度的同时能够更快的识别检测出目标。