← 返回列表
一种基于特征对齐与边缘约束的铁路场景分割方法
摘要文本
本发明公开了一种基于特征对齐与边缘约束的铁路场景分割方法，包括以下步骤：S1、获取铁路场景图像并构建场景分割模型；S2、将获取的图像送入特征提取网络中得到不同尺寸的特征图；S3、采用通道注意力机制对低维特征进行特征增强，对高维特征采用特征对齐网络实现可学习上采样；S4、将增强后的低维特征和上采样后的高维特征拼接后用于全局分割结果和边缘结果的预测；S5、利用全局分割结果和边缘分割结果进行损失函数计算，用于反向传播；S6、经过多轮迭代优化，得到最优的模型参数。本发明采用上述的一种基于特征对齐与边缘约束的铁路场景分割方法，可以对复杂的铁路场景实现全场景分割，提高场景中不同区域边缘的识别效果和整个场景的分割精度。
申请人信息
- 申请人:北京交通大学; 中车智能交通运营管理有限公司
- 申请人地址:100044 北京市海淀区西直门外上园村3号
- 发明人: 北京交通大学; 中车智能交通运营管理有限公司
专利详细信息
| 项目 | 内容 |
|---|---|
| 专利名称 | 一种基于特征对齐与边缘约束的铁路场景分割方法 |
| 专利类型 | 发明申请 |
| 申请号 | CN202410005965.3 |
| 申请日 | 2024/1/3 |
| 公告号 | CN117830633A |
| 公开日 | 2024/4/5 |
| IPC主分类号 | G06V10/26 |
| 权利人 | 北京交通大学; 中车智能交通运营管理有限公司 |
| 发明人 | 郭保青; 余祖俊; 李晓争; 朱力强; 孙涛 |
| 地址 | 北京市海淀区西直门外上园村3号; |
专利主权项内容
1.一种基于特征对齐与边缘约束的铁路场景分割方法，其特征在于，包括以下步骤：S1、获取铁路场景图像，并构建场景分割模型；场景分割模型中包括特征提取网络，特征对齐网络，边缘约束网络三部分；S2、将获取的铁路场景图像送入特征提取网络中得到不同尺寸的特征图；S3、采用通道注意力机制对低维特征进行特征增强，对高维特征采用特征对齐网络实现可学习上采样；S4、将增强后的低维特征和上采样后的高维特征拼接后采用全景预测模块和边缘预测模块对全局分割结果和边缘结果进行预测；S5、利用全局分割结果和全局分割标签，边缘分割结果和使用全局分割标签提取到的边缘标签进行损失函数计算，用于反向传播；S6、经过多轮迭代优化，得到最优的模型参数。 来自马-克-数-据