| 基于深度学习的SSD模型尾矿库自动提取 |
| |
| 作者姓名： | 闫凯 沈汀 陈正超 闫弘轩 |
| |
| 作者单位： | 1. 中国科学院遥感与数字地球研究所, 北京 100094;
2. 中国科学院大学电子电气与通信工程学院, 北京 100094 |
| |
| 基金项目： | 中国科学院A类战略性先导科技专项（XDA19080302）资助 |
| |
| 摘 要： | 针对华北地区尾矿库自动提取问题，将基于深度学习的SSD目标检测模型应用于遥感图像尾矿库提取。首先标记华北地区2 000个样本，随机挑选1 500个作为训练样本，剩余样本作为测试样本，验证模型的检测精度。分析卷积层对应感受野与图像中尾矿库尺寸关系，发现原始SSD模型漏检误检大型尾矿库。改进SSD模型结构，提出增加额外卷积层的策略，提高对大型尾矿库目标的检测精度。实验表明，在置信度阈值为0.3时，改进的SSD模型相比原始模型，检测精确率提高10.0%，召回率提高14.4%，提高了大型尾矿库检测精度。验证了基于深度学习的SSD目标检测模型自动提取尾矿库的可行性以及改进算法的有效性。
|
| 关 键 词： | 深度学习 目标检测 尾矿库 SSD模型 感受野 |
| 收稿时间： | 2019-01-02 |
| 修稿时间： | 2019-03-27 |
Automatic extraction of tailing pond based on SSD of deep learning |
| |
| Authors: | YAN Kai SHEN Ting CHEN Zhengchao YAN Hongxuan |
| |
| Institution: | 1. Institute of Remote Sensing and Digital Earth, Chinese Academy of Sciences, Beijing 100094, China;
2. School of Electronic Electrical and Communication Engineering, University of Chinese Academy of Sciences, Beijing 100094, China |
| |
| Abstract: | In order to automatically extract tailing ponds in North China, the SSD target detection model based on deep learning is applied to extract tailing ponds from remote sensing images. Firstly, 2 000 samples in North China were labeled as the foundation of database. 1 500 samples were randomly selected as training samples, and the remaining samples were used as test samples to verify the detection accuracy of the model. By using the original SSD model, the targets of large tailing ponds can not be detected accurately. In this work the relationship between the corresponding receptive field of convolution layer and the size of tailing reservoir in image was analyzed. Moreover, in order to improve the detection accuracy of the model for large-scale tailing pond targets, we modified the structure of SSD model by introducing an extra convolution layer. Experiments show that, compared with the original SSD model, the modified SSD model improves the detection accuracy by 10% and the recall rate by 14.4% at 0.3 confidence level. The detection accuracy for large tailing ponds is also improved. In this work, the feasibility of automatic extraction of tailing reservoir based on deep learning SSD model and the effectiveness of the modified algorithm were verified. |
| |
| Keywords: | deep learning object detection tailing pond SSD model receptive field |
|
| 点击此处可从《》浏览原始摘要信息 |
| 点击此处可从《》下载免费的PDF全文 |
|
