用MRNet模型提升嫦娥五号月球岩石分类精度。
Using MRNet to Predict Lunar Rock Categories Detected by Chang'e 5 Probe
- 基于VGG16与空洞卷积的U-Net结构,融合全局信息
- 在自建数据集上识别准确率达40.0%以上
- 适合月球表面岩石精细分类研究者使用
中国嫦娥五号任务成功登陆风暴洋区域,采集月面图像。过去半个世纪带回的月岩样本数量有限,当前主要依赖月球车对地表岩石进行探测分析。风暴洋区域岩石类型多样,本文向中国科学院国家天文台申请了着陆器导航与地形相机(NaTeCam)图像,构建了月面岩石图像数据集CE5ROCK,包含100张图像,并随机划分为训练、验证和测试集。实验表明,传统CNN模型如AlexNet或MobileNet在该数据集上的识别准确率约为40.0%。为更好利用月面图像中的全局信息,本文提出MRNet(MoonRockNet)网络架构:编码部分采用VGG16提取特征,解码部分在原始VGG16结构基础上引入空洞卷积与通用的U-Net结构,更有利于识别种类更细、分布更稀疏的月球岩石。在CE5ROCK数据集上开展大量实验,结果表明MRNet在岩石类型识别性能上优于现有主流算法。
原文摘要 · Abstract (English)
China's Chang'e 5 mission has been a remarkable success, with the chang'e 5 lander traveling on the Oceanus Procellarum to collect images of the lunar surface. Over the past half century, people have brought back some lunar rock samples, but its quantity does not meet the need for research. Under current circumstances, people still mainly rely on the analysis of rocks on the lunar surface through the detection of lunar rover. The Oceanus Procellarum, chosen by Chang'e 5 mission, contains various kind of rock species. Therefore, we first applied to the National Astronomical Observatories of the China under the Chinese Academy of Sciences for the Navigation and Terrain Camera (NaTeCam) of the lunar surface image, and established a lunar surface rock image data set CE5ROCK. The data set contains 100 images, which randomly divided into training, validation and test set. Experimental results show that the identification accuracy testing on convolutional neural network (CNN) models like AlexNet or MobileNet is about to 40.0%. In order to make full use of the global information in Moon images, this paper proposes the MRNet (MoonRockNet) network architecture. The encoding structure of the network uses VGG16 for feature extraction, and the decoding part adds dilated convolution and commonly used U-Net structure on the original VGG16 decoding structure, which is more conducive to identify more refined but more sparsely distributed types of lunar rocks. We have conducted extensive experiments on the established CE5ROCK data set, and the experimental results show that MRNet can achieve more accurate rock type identification, and outperform other existing mainstream algorithms in the identification performance.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。