MarsNet: Automated Rock Segmentation With Transformers for Tianwen-1 Mission
IEEE Geoscience and Remote Sensing Letters(2023)
摘要
The Mars exploration mission of China named Tianwen-1 is being carried out as scheduled. The Navigation and Terrain Cameras (NaTeCam) equipped on the Zhurong Rover play an essential role in obstacle recognition. The main obstacles on the Martian surface are rocks of different sizes, which influence the path planning of Zhurong Rover in scientific exploration. Most existing semantic segmentation methods are based on the U-Net architecture with ResNet or other backbones, and features extracted by these methods lack long-range dependencies. To fully exploit the context information, we propose the MarsNet framework for the Mars image, which combines transformers with the convolutional neural network (CNN) as the backbone, and hybrid dilated convolution (HDC) is also employed to the decoder path to help detect the huge rocks. Besides, since there are few open-source datasets for rock segmentation for Mars, we establish a segmentation dataset from the Martian surface image, named TWMARS, captured by NaTeCam. Extensive experiments are conducted on the TWMARS dataset, and the experimental results demonstrate that MarsNet achieves accurate rock segmentation and outperforms state-of-the-art methods. The source code is available at https://github.com/BUPT-ANT-1007/MarsNet.
更多查看译文
关键词
Hybrid dilated convolution (HDC),rock segmentation,transformers,TWMARS dataset,Zhurong Rover
AI 理解论文
溯源树
样例
生成溯源树,研究论文发展脉络
Chat Paper
正在生成论文摘要