一种基于深度卷积神经网络的宅基地目标检测方法技术

技术编号:27276215 阅读:24 留言:0更新日期:2021-02-06 11:42
本发明专利技术公开了一种基于深度卷积神经网络的宅基地目标检测方法,该方法先收集宅基地遥感图像数据;然后对原始图像数据进行预处理,主要是将原始图像进行人工标注,并使用翻转和旋转的方法进行数据扩充并进行了透视矫正;本发明专利技术设计一种Faster R

【技术实现步骤摘要】
一种基于深度卷积神经网络的宅基地目标检测方法


[0001]本专利技术涉及图像处理和计算机视觉
,具体涉及一种基于深度卷积神经网络的宅基地目标检测方法。

技术介绍

[0002]农村宅基地识别是观察农村的土地变化和经济发展的重要手段。近年来,高分辨率遥感影像开始覆盖人们生活的各个方面,不再只针对发达的城市地区。识别这些宅基地,对农村土地利用情况进行研究,则是新农村建设的基础。中国作为传统的农业大国,农村发展至关重要,快速、准确地获得农村宅基地的信息对于统筹城乡发展和节约集约用地有着重大的帮助。农村宅基地识别对于履行农村宅基地改革与管理职责、全面掌握农村宅基地现状、保护农民权益具有十分重要的意义。而目标检测是计算机视觉以及人工智能领域的一个基本问题,其主要目标是精确定位出图像中各种目标的类别和位置边框信息,可将该技术用于宅基地监管、监测和调查工作。
[0003]从上世纪90年代到本世纪2010年之前,传统的机器学习算法在遥感图像中的目标检测发挥了很大的作用,绝大多数的目标检测任务都由传统的机器学习算法完成。首先,依据特定的目标设计特定的特征提取算法,然后再将得到的特征向量送入选定的特征分类器进行训练。但是传统的机器学习算法特征提取困难,需要花大量的时间来设计较为合理的特征提取算法。其次,传统机器学习算法中使用的分类器如SVM网络结构泛化能力差,不具有表达复杂函数的能力。传统的分类过程存在使用门槛高、数据处理过程繁琐、识别速度慢、准确率不高和可迁移性差等弊端。
[0004]目前,深度卷积神经网络以其优良的性能,在图像处理领域的应用发展迅速,并且取得了很好的进展,而基于深度卷积神经网络的目标检测算法对遥感图像中的一个或多个目标的类别和位置进行自动化判定与识别,并以矢量框的形式作为标记,确定其类别和位置。然而传统的目标检测方法所使用的滑动窗口方法或者候选窗口方法效率较低,运算复杂度高,检测速度慢,并且依赖手工设计的特征,到了后期发展便出现瓶颈,难以在速度和精度上取得更大的突破。

技术实现思路

[0005]针对现有技术中的上述不足,本专利技术提供的一种基于深度卷积神经网络的宅基地目标检测方法解决了通过传统的分类过程存在使用门槛高、识别速度慢、准确率不高的问题。
[0006]为了达到上述专利技术目的,本专利技术采用的技术方案为:一种基于深度卷积神经网络的宅基地目标检测方法,包括以下步骤:
[0007]S1、通过无人机采集宅基地遥感图像集;
[0008]S2、对宅基地遥感图像集进行预处理,构建训练集、验证集和待识别的宅基地遥感图像;
[0009]S3、构建Faster R-CNN网络模型,采用训练集对Faster R-CNN网络模型进行训练,采用验证集对参数进行调整,得到训练完成的Faster R-CNN网络模型;
[0010]S4、采用训练完成的Faster R-CNN网络模型对待识别的宅基地遥感图像进行检测,得到宅基地目标检测结果,实现对遥感图像宅基地检测。
[0011]进一步地,步骤S2包括以下分步骤:
[0012]S21、对宅基地遥感图像集中所有图像进行裁剪,得到第一宅基地遥感标准图像集;
[0013]S22、将第一宅基地遥感标准图像集进行筛选,得到含宅基地目标且该目标位于裁剪分割线内的第二宅基地遥感标准图像集;
[0014]S23、对第二宅基地遥感标准图像集进行标注处理,得到宅基地遥感标注图像集;
[0015]S24、对宅基地遥感标注图像集进行扩充处理,得到宅基地遥感扩充图像集;
[0016]S25、对宅基地遥感扩充图像集进行透视矫正,得到样本图像数据集;
[0017]S26、将样本图像数据集划分为训练集、验证集和待识别的宅基地遥感图像。
[0018]进一步地,步骤S23包括以下分步骤:
[0019]S231、对第二宅基地遥感标准图像集中每张图像创建对应的xml格式文件;
[0020]S232、对xml格式文件标注宅基地物体的标签、图像尺寸和宅基地物体的bounding box坐标,得到宅基地遥感标注图像集;
[0021]所述步骤S24具体为:以90
°
、180
°
和270
°
三种角度对宅基地遥感标注图像集进行旋转和翻转,使bounding box的边界与坐标轴平行,得到宅基地遥感扩充图像集。
[0022]上述进一步方案的有益效果为:原有宅基地遥感图像数据集通过标注处理,和使用翻转和旋转的方法进行数据增强并进行了透视矫正,扩大了样本规模,更能模拟真实情况,同时提升了算法中特征提取对于不同形态的样本的表征能力,数据集的预处理使得Faster R-CNN网络模型训练更加准确。
[0023]进一步地,步骤S3中Faster R-CNN网络模型包括:特征提取网络、区域建议网络和目标定位网络;
[0024]所述特征提取网络的输入端作为Faster R-CNN网络模型的输入端,其输出端分别与目标定位网络的第一输入端和区域建议网络的输入端连接;
[0025]所述区域建议网络的输出端与目标定位网络的第二输入端连接;
[0026]所述目标定位网络的输出端作为Faster R-CNN网络模型的输出端。
[0027]进一步地,目标定位网络包括:ROI池化层、全连接层、softmax分类模块和第一边框回归模块;
[0028]所述ROI池化层的第一输入端作为目标定位网络的第一输入端,其第二输入端作为目标定位网络的第二输入端,其输出端与全连接层的输入端连接;
[0029]所述全连接层的输出端分别与softmax分类模块的输入端和第一边框回归模块的输入端连接;
[0030]所述softmax分类模块的输出端和第一边框回归模块的输出端共同作为目标定位网络的输出端。
[0031]进一步地,区域建议网络包括:滑动窗口模块、分类模块、第二边框回归模块和前景框模块;
[0032]所述滑动窗口模块的输入端作为区域建议网络的输入端,其输出端分别与分类模块的输入端和第二边框回归模块的输入端连接;
[0033]所述前景框模块的输入端分别与分类模块的输出端和第二边框回归模块的输出端连接,其输出端作为区域建议网络的输出端;
[0034]所述滑动窗口模块中的锚点数量为16个,具体为:64
×
64、128
×
128、256
×
256、512
×
512,其依次对应的比例大小为:3∶2、1∶2、1∶1、2∶1。
[0035]上述进一步方案的有益效果为:由于宅基地遥感图像具有多尺度的特点,且大部分宅基地在图像中的尺度较小,增加一组(64
×
64)的大小和(3∶2)的比例,用来提高小尺度物体的检测效率。
[0036]进一步地,步骤S3中Faster R-CNN网络模型中的特征提取网络、区域建议网络和目标定位网络分本文档来自技高网
...

【技术保护点】

【技术特征摘要】
1.一种基于深度卷积神经网络的宅基地目标检测方法,其特征在于,包括以下步骤:S1、通过无人机采集宅基地遥感图像集;S2、对宅基地遥感图像集进行预处理,构建训练集、验证集和待识别的宅基地遥感图像;S3、构建Faster R-CNN网络模型,采用训练集对Faster R-CNN网络模型进行训练,采用验证集对参数进行调整,得到训练完成的Faster R-CNN网络模型;S4、采用训练完成的Faster R-CNN网络模型对待识别的宅基地遥感图像进行检测,得到宅基地目标检测结果,实现对遥感图像宅基地检测。2.根据权利要求1所述的基于深度卷积神经网络的宅基地目标检测方法,其特征在于,所述步骤S2包括以下分步骤:S21、对宅基地遥感图像集中所有图像进行裁剪,得到第一宅基地遥感标准图像集;S22、将第一宅基地遥感标准图像集进行筛选,得到含宅基地目标且该目标位于裁剪分割线内的第二宅基地遥感标准图像集;S23、对第二宅基地遥感标准图像集进行标注处理,得到宅基地遥感标注图像集;S24、对宅基地遥感标注图像集进行扩充处理,得到宅基地遥感扩充图像集;S25、对宅基地遥感扩充图像集进行透视矫正,得到样本图像数据集;S26、将样本图像数据集划分为训练集、验证集和待识别的宅基地遥感图像。3.根据权利要求2所述的基于深度卷积神经网络的宅基地目标检测方法,其特征在于,所述步骤S23包括以下分步骤:S231、对第二宅基地遥感标准图像集中每张图像创建对应的xml格式文件;S232、对xml格式文件标注宅基地物体的标签、图像尺寸和宅基地物体的bounding box坐标,得到宅基地遥感标注图像集;所述步骤S24具体为:以90
°
、180
°
和270
°
三种角度对宅基地遥感标注图像集进行旋转和翻转,使bounding box的边界与坐标轴平行,得到宅基地遥感扩充图像集。4.根据权利要求1所述的基于深度卷积神经网络的宅基地目标检测方法,其特征在于,所述步骤S3中Faster R-CNN网络模型包括:特征提取网络、区域建议网络和目标定位网络;所述特征提取网络的输入端作为Faster R-CNN网络模型的输入端,其输出端分别与目标定位网络的第一输入端和区域建议网络的输入端连接;所述区域建议网络的输出端与目标定位网络的第二输入端连接;所述目标定位网络的输出端作为Faster R-CNN网络模型的输出端。5.根据权利要求4所述的基于深度卷积神经网络的宅基地目标检测方法,其特征在于,所述目标定位网络包括:ROI池化层、全连接层、softmax分类模块和第一边框回归模块;所述ROI池化层的第一输入端作为目标定位网络的...

【专利技术属性】
技术研发人员:李潇熊洋
申请(专利权)人:空间信息产业发展股份有限公司
类型:发明
国别省市:

网友询问留言 已有0条评论
  • 还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。

1