一种基于Transformer的宅基地遥感图像违建识别方法及系统与流程

一种基于transformer的宅基地遥感图像违建识别方法及系统
技术领域
1.本发明涉及g06k9/00领域，具体为一种基于transformer的宅基地遥感图像违建识别方法及系统。


背景技术：

2.目前，宅基地违规建设的识别方法主要是通过人工对高分辨率遥感图像的目测比较与标注。近年来，随着机器学习以及深度学习的发展，不少研究人员提出基于相关研究的方法，虽然取得了一定的效果，但是传统的人工对高分辨率遥感图像的目测比较与标注存在人力资源投入大、效率低下、费时费力等问题。
3.中国专利cn111692978a公开了一种基于多源影像数据的违章建筑识别方法，主要通过卫星图像识别并根据建筑物的本地太阳高度角信息以及建筑物的影子，求解获得建筑物的实际占地面积以及建筑物高度，并与备案信息相对比识别，虽然可以应用于城市建筑违章识别中，但是这种基于像素特征的统计学习或者机器学习方法存在过程复杂，容易因为输入照片特征的变化而出现精度低下的问题。中国专利cn109101984a公开了一种基于卷积神经网络的图像识别方法及装置，虽然一定程度提高了识别准确度，但是基于卷积神经网络的识别系统往往需要经过数次局部的卷积操作之后，经过堆叠的卷积层进行上采样和下采样，最后输出结果；然而农村违建宅基地在卫星或者无人机拍摄的遥感图像中占据很小的面积，多次下采样容易造成特征的丢失，严重影响识别的精度。
4.因此，提供一种基于transformer的宅基地遥感图像违建识别方法，充分利用计算机视觉与深度学习方法根据遥感图像精确识别与定位宅基地，解决传统方法下精确度低、人力资源投入大、效率低下、费时费力等问题，满足农村宅基地违建识别领域的实际应用要求，具有重要的现实研究和应用意义。


技术实现要素：

5.为了解决上述问题，本发明提供了一种基于transformer的宅基地遥感图像违建识别方法及系统，通过利用transformer作为骨干网络的全局自注意力机制捕获需要重点关注的区域，即遥感中的宅基地，最后结合faster rcnn范式进行目标检测与标注。
6.本发明一方面提供了一种基于transformer的宅基地遥感图像违建识别方法，至少包括以下步骤：
7.步骤1：获取农村宅基地的遥感图像；
8.步骤2：通过获取的遥感图像数据，构建数据集并对数据集进行划分；
9.步骤3：图像预处理与图像增强；
10.步骤4：将经过预处理与增强的图像输入到transformer模型中进行全局自注意力特征提取得到特征图；
11.步骤5：采用rpn网络通过特征图提出区域提案；
12.步骤6：收集特征图和区域提案，提取提案特征图；
13.步骤7：将提案特征图送入分类器进行分类并且得到候选框
14.步骤8：构建基于transformer的宅基地遥感图像违建识别系统。
15.作为一种优选的技术方案，所述步骤1中获取农村宅基地的遥感图像的的方式包括卫星拍摄、航空拍摄以及近景拍摄，可以充分适应不同地区农村的实际情况，结合卫星拍摄、航空拍摄以及近景拍摄以提供全方位、高准确度、高清晰度的农村宅基地遥感图像。
16.作为一种优选的技术方案，所述步骤2中数据集在机器学习/深度学习流程中进行划分，划分为训练集和测试集；训练集的作用是模型的训练，给予模型输入特征以及真实值，让模型学习其对应关系，在训练的过程中，训练集随机划分得到验证集，所述训练集与验证集的比例为9：1，作用是在模型对训练集进行一次完整的训练之后，验证模型的收敛程度；测试集的作用是在训练结束之后，测试模型的拟合程度。
17.作为一种优选的技术方案，所述步骤3中图像预处理与图像增强方式至少包括：
18.(1)将图像重新调整到符合模型输入的大小；
19.(2)对图像进行几何变换；
20.(3)对图像进行标准化和归一化；
21.(4)对图像的属性进行随机调整；
22.(5)对图像进行随机的遮挡。
23.优选的，所述方式2中几何变换至少包括平移、翻转和旋转；优选的，所述方式4中随机调整至少包括亮度调整、对比度调整。
24.本技术在设计及探究过程中发现，通过采用上述方式进行图像预处理和图像增强，使输入的遥感图像符合transformer模型的要求，另一方面提高了transformer模型的泛化能力，便于后续特征提取。
25.作为一种优选的技术方案，所述步骤4中transformer模型具体为swin transformer模型，采用分层和滑窗的设计，引入了计算机视觉先验的局部操作，在不丢失全局信息的同时，更好的提取局部的信息，因此也更加适用于农房宅基地目标检测的任务。
26.作为一种优选的技术方案，所述swin transformer模型经过imagenet-22k预训练的版本进行微调迁移学习，以求更快的收敛以及达到更好的效果。优选的，在后续的训练中，我们采用adamw优化器训练模型进行训练；
27.作为一种优选的技术方案，所述swin transformer模型提取特征的过程包括：输入224*224*1的图像，经过四层特征提取块的提取得到最终的特征图，所述特征图的尺寸分别为56*56*4、28*28*8、14*14*16、7*7*32。优选的，所述四层特征提取块中每一层中堆叠数个swin自注意力模块和多层感知机，输入的图像经过swin自注意力模块会通过滑窗操作将图片进行分窗、自注意力机制计算、移位和重新组合，通过引入计算机视觉先验，即层次性、局部性和平移不变性，更加适用于目标检测的任务中的特征提取。输入的图像向量与参数权重矩阵wq，wk，wv相乘得到三个矩阵q，v，k，最后再通过计算得到最后的注意力特征。优选的，所述swin自注意力模块中引入一个相对位置偏置b，进一步加强了swin transformer模型的泛化性能。
28.所述自注意力机制算法公式如下：
[0029][0030]
其中，attention(q，k，v)为自注意力机制的输出结果。q为查询(query)矩阵，k为键值(key)矩阵，d为k的维度，t表示转置，b表示相对位置偏置，v表示真值(value)矩阵，softmax为激活函数。
[0031]
本技术中采用的多层感知机是简单高效的前馈人工神经网络模型，将输入的多个数据集通过前馈人工神经网络模型映射到单一数据集上进行输出。
[0032]
作为一种优选的技术方案，所述步骤5中采用rpn网络通过特征图提出区域提案具体分为路径1和路径2；所述路径1为通过softmax激活函数分类锚点获得positive和negative分类；所述路径2为用于计算对应锚点的边框回归偏移量；通过提案层综合positive锚点和对应边框回归偏移量获取精确的提案，同时剔除太小和超出边界的提案，实现目标定位的功能。
[0033]
作为一种优选的技术方案，所述步骤6具体为：将通过swin transformer模型提取的特征图和采用rpn网络获得的区域提案输入roi pooling层，通过roi pooling层对特征信息进行综合后即可获得提案特征图，之后通过后续的全连接层对获得的提案特征图进行目标类别判定。
[0034]
所述步骤5、步骤6通过faster rcnn范式目标检测框架结合swin transformer模型有效提高目标提取、定位和判定精确度。
[0035]
作为一种优选的技术方案，所述步骤7中分类利用已经获得的提案特征图，通过全连接层与softmax激活函数计算并区分每个提案具体是否为农房宅基地，输出概率向量；为了获得精确的农房宅基目标检测框，需要采用边框回归对每个提案的位置偏移量进行再次确定和修正。
[0036]
作为一种优选的技术方案，所述步骤8中基于transformer的宅基地遥感图像违建识别系统的构建通过训练好的模型结合差异图计算算法进行构建。
[0037]
本发明另一方面提供了一种基于transformer的宅基地遥感图像违建识别系统，所述系统至少包括遥感图像获取与输入模块，图像预处理模块，目标检测模块，目标相似度计算模块，预测结果模块。
[0038]
所述遥感图像获取与输入模块用于当前宅基地遥感图像的获取并且将其输入到系统之中。
[0039]
所述图像预处理模块用于输入的遥感图像用与模型训练相同的图像预处理方式对图像进行预处理，确保模型提取的特征可以正确的用于目标检测。
[0040]
所述目标检测模块用于将经过预处理的遥感图像通过训练好的模型检测途中所有的农房宅基地。
[0041]
所述目标相似度计算模块用于与过去采集的农房宅基地遥感图像进行差异图相似度计算。
[0042]
所述预测结果模块用于根据目标相似度计算模块中计算的差异图相似度判断是否具有宅基地违建的情况，并且输出最终的识别结果。
[0043]
有益效果：
[0044]
1、本发明提供了一种基于transformer的宅基地遥感图像违建识别方法及系统，
解决了传统方法识别准确度不足，人力资源投入大，效率低下，费时费力等问题。
[0045]
2、本发明通过利用计算机视觉与深度学习方法根据遥感图像精确的识别与定位宅基地，避免识别结果受到遥感图像尺度与检测目标大小变化的影响，尤其适用于农房宅基地的准确识别。
[0046]
3、通过利用transformer作为骨干网络的全局自注意力机制捕获农房宅基地，结合faster rcnn范式进行目标检测与标注，有效提高目标提取、定位和判定精确度。
[0047]
4、采用本技术提供的基于transformer的宅基地遥感图像违建识别方法及系统，与现有数据库中的过去采集的农房宅基地遥感图像进行差异图相似度计算，可以准确判断是否为违规建筑。
[0048]
5、本技术提供的基于transformer的宅基地遥感图像违建识别方法及系统，为农村违建宅基地的快速准确识别提供了一种新的技术手段，有助于相关管理部门提高工作效率，具有很高的实际应用价值。
附图说明
[0049]
图1为本发明基于transformer的宅基地遥感图像违建识别方法中模型训练的流程图。
[0050]
图2为本发明基于transformer的宅基地遥感图像违建识别系统的结构示意图。
具体实施方式
[0051]
请参阅图1，图1为本发明基于transformer的宅基地遥感图像违建识别方法中模型训练的流程图。
[0052]
s10：获取农村宅基地的遥感图像。
[0053]
获取农村宅基地的方式为卫星拍摄、航空拍摄以及近景拍摄。
[0054]
s20：通过获取的遥感图像数据，构建数据集，划分训练集和测试集。
[0055]
s30：图像预处理与图像增强。
[0056]
图像预处理与图像增强方式为：
[0057]
(1)将图像重新调整到符合模型输入的大小；
[0058]
(2)对图像进行平移、翻转和旋转几何变换；
[0059]
(3)对图像进行标准化和归一化；
[0060]
(4)对图像的属性(亮度和对比度)进行随机调整；
[0061]
(5)对图像进行随机的遮挡。
[0062]
s40：将经过预处理与增强的图像输入到transformer模型中进行全局自注意力特征提取。
[0063]
采用swin transformer模型，所述swin transformer模型经过imagenet-22k预训练的版本进行微调迁移学习，在后续的训练中采用adamw优化器训练模型，初始学习率设置为0.0001、权重衰减设置为0.05以及批次大小设置为16；每训练完一批次为一个迭代，每进行s20中的一次完整训练即用验证集进行一次验证并输出实时的精度；设置36次完整训练即结束训练，最后用测试集检测模型的拟合程度。
[0064]
swin transformer模型提取特征的过程包括：输入224*224*1的图像在经过四层
特征提取块的提取得到最终的特征图，其尺寸分别为56*56*4、28*28*8、14*14*16、7*7*32。
[0065]
四层特征提取块中每一层中堆叠数个swin自注意力模块和多层感知机，输入的图像经过swin自注意力模块会通过滑窗操作将图片进行分窗、自注意力机制计算、移位和重新组合，通过引入计算机视觉先验，即层次性、局部性和平移不变性。
[0066]
输入的图像向量与参数权重矩阵wq，wk，wv相乘得到三个矩阵q，v，k，最后再通过计算得到最后的注意力特征。
[0067]
优选的，所述swin自注意力模块中引入一个相对位置偏置b，进一步加强了swin transformer模型的泛化性能。
[0068]
所述自注意力机制算法公式如下：
[0069][0070]
其中，attention(q，k，v)为自注意力机制的输出结果。q为查询(query)矩阵，k为键值(key)矩阵，d为k的维度，t表示转置，b表示相对位置偏置，v表示真值(value)矩阵，softmax为激活函数。
[0071]
本技术中采用的多层感知机是简单高效的前馈人工神经网络模型，将输入的多个数据集通过前馈人工神经网络模型映射到单一数据集上进行输出。
[0072]
s50：rpn网络通过特征图提出区域提案；
[0073]
采用rpn网络通过特征图提出区域提案具体分为路径1和路径2；所述路径1为通过softmax激活函数分类锚点获得positive和negative分类；所述路径2为用于计算对应锚点的边框回归偏移量；通过提案层综合positive锚点和对应边框回归偏移量获取精确的提案，同时剔除太小和超出边界的提案，实现目标定位的功能。
[0074]
s60：收集特征图和区域提案，提取提案特征图；
[0075]
将通过swin transformer模型提取的特征图和采用rpn网络获得的区域提案输入roi pooling层，通过roi pooling层对特征信息进行综合后即可获得提案特征图，之后通过后续的全连接层对获得的提案特征图进行目标类别判定。
[0076]
s70：将提案特征图送入分类器进行分类并且得到候选框；
[0077]
分类部分利用已经获得的提案特征图，通过全连接层与softmax激活函数计算并区分每个提案具体是否为农房宅基地，输出概率向量；为了获得精确的农房宅基目标检测框，采用边框回归对每个提案的位置偏移量进行再次确定和修正。
[0078]
请参阅图2，图2为本发明基于transformer的宅基地遥感图像违建识别系统的结构示意图。
[0079]
所述基于transformer的宅基地遥感图像违建识别系统的结构系统包括遥感图像获取与输入模块100、图像预处理模块200、目标检测模块300、目标相似度计算模块400、预测结果模块500。
[0080]
遥感图像获取与输入模块100用于当前宅基地遥感图像的获取并且将其输入到系统之中。
[0081]
图像预处理模块200用于输入的遥感图像用与模型训练相同的图像预处理方式对图像进行预处理，确保模型提取的特征可以正确的用于目标检测。
[0082]
目标检测模块300用于将经过预处理的遥感图像通过训练好的模型检测途中所有
的农房宅基地。
[0083]
目标相似度计算模块400用于与过去采集的农房宅基地遥感图像进行差异图相似度计算。
[0084]
预测结果模块500用于根据400中计算的差异图相似度判断是否具有宅基地违建的情况，并且输出最终的识别结果。