利索能及
我要发布
收藏
专利号: 2022108524601
申请人: 浙江工业大学
专利类型:发明专利
专利状态:已下证
更新日期:2026-08-19
缴费截止日期: 暂无
联系人

摘要:

权利要求书:

1.一种融合空间关系的高分遥感目标检测方法,包括以下步骤:步骤1:制作样本,具体包括:

步骤1.1:获取高分遥感影像:采用具有可见光‑近红外传感器的光学卫星遥感数据或搭载一般光学相机的航空遥感数据,根据分辨率要求可直接使用多光谱影像或融合全色影像;

步骤1.2:根据遥感影像大小裁剪:在生产区域选择典型目标所在范围将遥感影像统一裁剪成相同像素大小;

步骤1.3:制作道路提取模型训练样本;

步骤1.4:根据任务需求一般采集数量较多的影像及相应标注作为样本,按照比例划分样本为训练集和测试集;

步骤2:训练道路提取模型,具体包含:

步骤2.1:设计深度卷积神经网络,选取适用于提取目标的深度学习网络结构,并对其改进,根据生产目标网络可相应替换、修改;

步骤2.2:初始化权重:使用已有的网络模型当作预训练模型来初始化网络权重;

步骤2.3:设置训练超参数:配置模型调优后具体数值;

步骤2.4:输入样本,训练模型:将训练样本输入到网络中,按照步骤2.3中所述的超参数进行训练,得到能够提取道路信息的道路提取模型;

步骤3:获取道路信息作为“先验知识”,具体包括:

步骤3.1:将高分遥感影像输入到道路提取模型,预测每个像素是道路的可能性,输出形成从0至255的灰度图,调整阈值将道路提取图转化为二值图,转换公式如下:其中threshold为区间[0,1]上的实数,可由用户设置;x,y为图像的横纵坐标;

步骤3.2:以道路为中心,辐射三个像素点作为缓冲区,提取缓冲区范围;

步骤4:制作目标检测网络模型训练样本,具体包括:

步骤4.1:遥感图像采用与道路提取模型所裁剪的一致;

步骤4.2:制作目标检测网络模型训练样本与标签;

步骤5:训练目标检测模型,具体包括:

步骤5.1:设定好训练目标检测模型的训练参数;

步骤5.2:图像传入网络,在输入端进行数据增强;在Backbone部分数据切片,提高检测速度;Neck部分采用借鉴CSP2结构,加强网络特征融合的能力;

步骤5.3:经过前面的网络,可以得到一张特征图,将特征图分成若干个网格(grid cell),网格划分的大小由输入端图像大小决定,如果某个目标的中心落在这个网格中,则这个网格就负责预测这个目标;每个网络需要预测三个目标框的位置信息和置信度(confidence)信息,一个目标框对应着四个位置信息和一个置信度信息;置信度代表所预测的目标框中包含目标的置信度和这个目标框预测的准确度这两重信息;

步骤5.4:目标中心落在一个网格里,记录该目标的各类信息,输出它的张量tensor;

步骤5.5:在预测过程中采用道路位置作为“先验知识”,由于在本实例中主要检测地物目标为服务区、收费站,主要分布于道路两侧,有着明显的空间相邻关系,所以根据道路二值图形成的缓冲区划定范围,只在道路辐射三格的区域进行筛选,减少对干扰因素的预测,提高检测速度;

步骤5.6:现已经yolo网络得到尺度特征图,尺度特征图上每个二维点,对应输入图像中的网格,关联特征图与预测框,开启端到端训练,经线性回归计算,使得锚框接近目标的真实标签;

步骤6:预测值经过损失,具体包括:

步骤6.1:在训练过程中,使用目标检测网络模型进行预测,并分别计算三个损失:置信度损失、类别损失、框位置损失;根据作者所提出的边框回归公式不断矫正预测框与标签框的重合度;

步骤6.2:再与道路二值图同传入本发明的道路损失进行回归计算,使网络学习到检测的目标靠近道路附近;

步骤7:遍历图像,录入矢量信息,具体包括:

步骤7.1:参数化外包框:将目标检测模型输出的矩形外包框参数转化为其左下顶点坐标和右上顶点坐标;转化公式如下:步骤7.2:原有的像素点坐标转为空间地理坐标,遍历图像,得到精度高的预测框,对比原始方法的预测框可以发现,本发明有比较明显的精度优势;写入矢量文件,写入过程判断矢量文件是否存在,若存在正常写入,若不存在,先创建矢量文件再写入;再连接四个空间地理坐标,使其闭合。

2.根据权利要求1所述的融合空间关系的高分遥感目标检测方法,其特征在于:步骤2训练道路提取模型的超参数,具体包括:DLinkNet的训练参数设定:迭代次数=400,batch_size=4,学习率更新策略=step,学习率更新步长=[100,200,300,400],初始学习率=0.0002,学习率更新系数=0.2;

步骤5训练道路提取模型的超参数,具体包括:

Yolo的训练参数设定:迭代次数=300,batch_size=32,学习率更新策略=step,学习率更新步长=[3200,4800,6400,8000],初始学习率=0.001,学习率更新系数=0.1。

3.根据权利要求1所述的融合空间关系的高分遥感目标检测方法,其特征在于:步骤1制作样本过程中,可选择空间关系较为明确的地物,空间关系发挥的作用越大,对网络的训练帮助效果越大,目标检测的效果越好。

4.根据权利要求1所述的融合空间关系的高分遥感目标检测方法,其特征在于:步骤1将遥感影像统一裁剪成1280*1280像素大小;步骤五将遥感影像统一填充成640*640像素大小。

5.根据权利要求1所述的融合空间关系的高分遥感目标检测方法,其特征在于:步骤5所进行的预测过程,把输入图像划分成Sh*Sw个网格,每个网格要预测B个目标框,还要预测C个类别;输出为Sh*Sw*(5*B+C)的张量tensor。