arXiv:2509.08027cs.CVcs.LG2025-09被引 1

MCTED是用于火星地形建模的开源机器学习数据集,包含8万余张影像与高程图配对样本。

MCTED: A Machine-Learning-Ready Dataset for Digital Elevation Model Generation From Mars Imagery

  • 基于火星勘测轨道器图像构建,含80,898个样本,覆盖广泛地貌
  • 自研工具修复原始数据缺陷,确保训练无偏差
  • 小模型在该数据集上表现优于零样本深度估计模型

本文提出一个面向火星数字高程模型预测任务的机器学习友好型数据集MCTED。该数据集通过处理来自Day等人的高分辨率火星正射影像与DEM配对数据生成,共包含80,898个数据样本。原始数据来源于火星勘测轨道器的CTX仪器,覆盖火星表面多样地貌。针对大规模DEM生成中常见的伪影与缺失数据问题,我们开发了专用工具进行修复或缓解影响。数据按训练与验证集划分,且两集样本不重叠,避免数据泄露。每个样本包含光学影像块、高程图块及两个掩码块(分别标记原始缺失值和人工修改区域),便于后续用户灵活处理异常高程区。我们还提供了数据的空间分布、高程值、坡度等统计信息。最后,在MCTED上训练小型U-Net模型,并与零样本深度估计模型DepthAnythingV2对比,发现该小模型性能优于后者。数据集与生成代码已完全开源。

原文摘要 · Abstract (English)

This work presents a new dataset for the Martian digital elevation model prediction task, ready for machine learning applications called MCTED. The dataset has been generated using a comprehensive pipeline designed to process high-resolution Mars orthoimage and DEM pairs from Day et al., yielding a dataset consisting of 80,898 data samples. The source images are data gathered by the Mars Reconnaissance Orbiter using the CTX instrument, providing a very diverse and comprehensive coverage of the Martian surface. Given the complexity of the processing pipelines used in large-scale DEMs, there are often artefacts and missing data points in the original data, for which we developed tools to solve or mitigate their impact. We divide the processed samples into training and validation splits, ensuring samples in both splits cover no mutual areas to avoid data leakage. Every sample in the dataset is represented by the optical image patch, DEM patch, and two mask patches, indicating values that were originally missing or were altered by us. This allows future users of the dataset to handle altered elevation regions as they please. We provide statistical insights of the generated dataset, including the spatial distribution of samples, the distributions of elevation values, slopes and more. Finally, we train a small U-Net architecture on the MCTED dataset and compare its performance to a monocular depth estimation foundation model, DepthAnythingV2, on the task of elevation prediction. We find that even a very small architecture trained on this dataset specifically, beats a zero-shot performance of a depth estimation foundation model like DepthAnythingV2. We make the dataset and code used for its generation completely open source in public repositories.

火星测绘深度学习数据集高程重建

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。