基于改进yolov4-tiny的地铁受电弓燃弧检测方法

地铁受电弓的燃弧会加剧受电弓滑板和接触网磨损,严重危害轨道交通安全。文章针对地铁车辆受电弓燃弧检测存在强光干扰和背景多变的问题,提出了一种基于yolov4-tiny模型改进的燃弧检测方法。为提升小目标检测能力,该方法在yolov4-tiny原有两个尺度的预测分支的基础上,添加第三尺度的预测分支以实现小燃弧在浅层网络中的定位,在主干网络后增加RFB(Receptive Field Block)模块以扩大网络的感受野,增强模型的特征提取能力。结果表明,改进的模型在测试集上的平均精度值(PAP)比yolov4-tiny提升了7.8个百分点,达到了98.2%,燃弧的定位效果与yolov4相当,但速度得到了极大的提升,单张图片的推理速度仅为6.5 ms,能有效、准确地完成地铁车辆中的受电弓燃弧检测任务。