发明公开
- 专利标题: 一种基于密集网络和注意力机制的铁路沿线遥感地物目标检测方法
-
申请号: CN202111315604.1申请日: 2021-11-08
-
公开(公告)号: CN113920435A公开(公告)日: 2022-01-11
- 发明人: 王阳萍 , 韩淑梅 , 杨景玉 , 党建武 , 雍玖 , 岳彪 , 王松 , 王文润 , 陈永 , 任鹏百 , 杨艳春
- 申请人: 兰州交通大学
- 申请人地址: 甘肃省兰州市安宁区安宁西路88号
- 专利权人: 兰州交通大学
- 当前专利权人: 兰州交通大学
- 当前专利权人地址: 甘肃省兰州市安宁区安宁西路88号
- 代理机构: 甘肃省知识产权事务中心代理有限公司
- 代理商 赵红红
- 主分类号: G06V20/10
- IPC分类号: G06V20/10 ; G06V10/774 ; G06V10/764 ; G06V10/82 ; G06N3/04 ; G06N3/08
摘要:
本发明公开了一种基于密集网络和注意力机制的铁路沿线遥感图像地物检测方法,涉及深度学习及遥感图像地物目标检测领域。包括使用DenseNet模块代替YOLOv4网络结构CSPDarknet53中部分CSP单元的ResNet残差模块,以实现特征重用;在YOLOv4的骨干网中的每个CSP_Unit中增加一个压缩激励结构,以增强提取特征的能力;在输出网络之前引入通道和空间注意力机制,从而提高检测的准确性。制作铁路沿线遥感地物目标检测数据集,并在该数据集上对改进后的YOLOv4网络结构进行训练,得到训练好的铁路沿线遥感地物检测模型,进行地物检测。本发明能够提高检测速度和检测精度,减小模型大小,适合铁路沿线遥感地物目标检测,满足实时性的要求。