发明公开
- 专利标题: 基于深度学习的地铁多色灯信号机识别方法、设备及介质
-
申请号: CN202310777482.0申请日: 2023-06-28
-
公开(公告)号: CN117011826A公开(公告)日: 2023-11-07
- 发明人: 张中坤 , 王思远
- 申请人: 卡斯柯信号有限公司
- 申请人地址: 上海市静安区江场路1401弄21号1101室
- 专利权人: 卡斯柯信号有限公司
- 当前专利权人: 卡斯柯信号有限公司
- 当前专利权人地址: 上海市静安区江场路1401弄21号1101室
- 代理机构: 上海科盛知识产权代理有限公司
- 代理商 应小波
- 主分类号: G06V20/58
- IPC分类号: G06V20/58 ; G06V20/40 ; G06V20/52 ; G06V10/26 ; G06V10/82 ; G06T7/90 ; G06N3/0464 ; G06N3/08
摘要:
本发明涉及一种基于深度学习的地铁多色灯信号机识别方法、设备及介质,该方法包括:步骤S1,车载相机实时监测并获取前方轨道线路的视频数据,构建色灯信号机数据集;步骤S2,将色灯信号机数据集输入深度学习网络训练后进行加速处理,得到检测色灯信号机的模型;步骤S3,根据视频数据采用深度学习语义分割网络识别出列车前方即将行驶的轨行区,得到含有轨行区识别结果的图像;步骤S4,将轨行区右侧的原始图像输入检测色灯信号机的模型,得到所有色灯位置的目标框;步骤S5,判断所有色灯的颜色,并选择距离列车最近的色灯;步骤S6,将距离列车最近的色灯位置和颜色实时打印输出。与现有技术相比,本发明具有提升了识别速度和精度等优点。