基于改进Mask R-CNN的矿区无人机影像沙棘识别方法及系统
摘要文本
本发明公开了一种基于改进Mask R‑CNN的矿区无人机影像沙棘识别方法及系统，其方法包括：S1、构建包括特征提取网络、注意力机制模块、RPN区域生成网络、检测头网络系统的改进Mask R‑CNN网络模型，特征提取网络对输入影像处理得到特征图P2～P6；S2、注意力机制模块对输入特征图加权处理得到特征图Q2～Q6；S3、RPN区域生成网络遍历特征图Q2～Q6每个像素点并得到候选目标区域RoIs；检测头网络系统经过识别与分割相结合的处理得到包含边界框、类别信息的特征图像；S4、采集研究区的正射影像输入训练后的改进Mask R‑CNN网络模型并得到所对应的特征图像。本发明提高了沙棘的识别效率和准确性，能够更可靠地捕捉和分析沙棘植被数据，为矿区沙棘的管理和维护提供有力支持。 微信公众号专利查询网
申请人信息
- 申请人:中国矿业大学(北京)
- 申请人地址:100083 北京市海淀区学院路丁11号
- 发明人: 中国矿业大学(北京)
专利详细信息
| 项目 | 内容 |
|---|---|
| 专利名称 | 基于改进Mask R-CNN的矿区无人机影像沙棘识别方法及系统 |
| 专利类型 | 发明申请 |
| 申请号 | CN202311378728.3 |
| 申请日 | 2023/10/24 |
| 公告号 | CN117351359A |
| 公开日 | 2024/1/5 |
| IPC主分类号 | G06V20/10 |
| 权利人 | 中国矿业大学(北京) |
| 发明人 | 张成业; 谢焱新; 李军; 杜守航; 邢江河 |
| 地址 | 北京市海淀区学院路丁11号中国矿业大学综合楼545 |
专利主权项内容
1.一种基于改进Mask R-CNN的矿区无人机影像沙棘识别方法，其特征在于：其方法包括：S1、构建沙棘正射影像样本集，沙棘正射影像样本集包括沙棘正射影像样本及对应的类别标签；基于Pytorch构建改进Mask R-CNN网络模型，改进Mask R-CNN网络模型以ResNet-50网络与FPN网络作为特征提取网络，ResNet-50网络对输入影像进行特征提取得到特征图C1～C5，然后经过1×1卷积操作输入FPN网络，FPN网络通过上采样、卷积处理得到融合上下层特征的特征图P2～P6；S2、改进Mask R-CNN网络模型在特征提取网络后端设有注意力机制模块，注意力机制模块对输入特征图进行长度、宽度方向的位置信息编码分成两个方向的特征向量，接着经过卷积调整两个方向特征向量的通道数，再经过Sigmoid函数处理并与输入特征图进行加权处理得到特征图Q2～Q6；S3、改进Mask R-CNN网络模型在注意力机制模块后端还设有检测头网络系统，在注意力机制模块与检测头网络系统之间还设有RPN区域生成网络，RPN区域生成网络遍历特征图Q2～Q6每个像素点并得到候选目标区域RoIs；检测头网络系统包括识别网络和分割网络，检测头网络系统将候选目标区域RoIs与特征图Q2～Q6进行对齐处理，识别网络通过一个卷积层和和两个全连接层得到目标的边界框和类别信息；分割网络产生与目标叶片大小、形状一致的掩膜对目标叶片图像进行分割，最后将识别与分割结果相结合得到包含边界框、类别及掩膜信息的特征图像，特征图像与输入特征图相对应；S4、向改进Mask R-CNN网络模型输入沙棘正射影像样本集进行模型训练，采集研究区的正射影像输入训练后的改进Mask R-CNN网络模型，输出得到正射影像所对应的特征图像，特征图像包含边界框、类别及掩膜信息。