SyMTRS
收藏数据集概述:SyMTRS
SyMTRS 是一个基于 Unreal Engine 5 的高保真仿真数据集,构建于 MatrixCity 项目(ArXiv)之上,提供具有精确几何、光照和传感器参数控制的逼真城市场景。该数据集为多模态、完美对齐的数据,包含高分辨率(2048×2048)的 RGB 航空图像。
主要特征
- 仿真管道:高保真度,基于 Unreal Engine 5。
- 场景类型:逼真的城市场景。
- 控制能力:可精确控制几何、光照和传感器参数。
- 数据模态:多模态,数据完美对齐。
- 图像分辨率:高分辨率 RGB 航空图像 (2048×2048)。
数据获取
数据集可通过 Hugging Face 仓库下载:
-
Git 克隆: sh git clone https://huggingface.co/datasets/safouaneelg/SyMTRS
-
Hugging Face
datasets库加载: python from datasets import load_dataset ds = load_dataset("safouaneelg/SyMTRS")
数据集布局
代码假定以下目录结构,文件通过文件名前缀匹配:
/path/to/SyMTRS/ ├── hr/ # 高分辨率图像 │ ├── RS.0.png │ ├── RS.100200.png │ └── ... ├── lr/ # 低分辨率图像 │ ├── x2/ # 2倍下采样 │ ├── x4/ # 4倍下采样 │ └── x8/ # 8倍下采样 ├── night/ # 夜间图像 │ ├── RS.0.png │ └── ... └── depth/ # 深度图 (numpy 格式) ├── RS.depth.0.npy └── ...
文件匹配规则:
- SuperResDataset:通过相同的文件名前缀匹配低分辨率和对应的高分辨率图像。
- DomainAdaptDataset:当
paired=True时,通过相同的文件名前缀匹配日间和夜间图像。 - DepthDataset:通过文件名前缀匹配深度文件(如
RS.depth.100200.npy)与对应的图像文件(如RS.100200.png)。
支持的任务与模型
该数据集仓库支持以下任务,并提供了相应的 PyTorch 数据加载器和训练流程:
-
超分辨率 (Super-resolution)
- 入口点:
python train_superresolution.py ... - 实现的模型:
srcnnautoencodersrganswinir
- 关键参数:
--lr_root(低分辨率图像目录)、--hr_root(高分辨率图像目录)、--model(模型名)、--scale(上采样倍数,支持2,4,8)、--batch_size(默认4)、--epochs(默认20)、--lr(默认1e-4)等。
- 入口点:
-
图像到图像翻译 (Image-to-image translation)
- 位置:CycleGAN-and-pix2pix/README.md
- 实现内容:
- CycleGAN:用于非成对的日间/夜间图像翻译。
- pix2pix:用于成对的日间/夜间图像翻译。
-
数据加载工具 (Dataset loading utilities)
- 提供以下数据加载器:
SuperResDataset:用于超分辨率任务。DomainAdaptDataset:用于域适应任务(日/夜)。DepthDataset:用于深度估计任务。
- 提供以下数据加载器:
环境配置
-
根项目:未提供环境文件,需手动安装依赖。 bash pip install torch torchvision numpy pillow matplotlib tqdm timm
timm为swinir模型所需。- CUDA 为可选。
-
图像翻译子项目:需使用其独立的运行环境(详见 CycleGAN-and-pix2pix/README.md)。
超分辨率训练输出
每次训练运行的结果保存至 --out_dir 参数指定的目录,包含:
metrics.csv:指标记录文件。loss_curve.png:损失曲线图。psnr_curve.png:PSNR 曲线图。mse_curve.png:MSE 曲线图。ssim_curve.png:SSIM 曲线图。weights/:模型权重文件(best.pt和last.pt)。samples/:训练过程中的样本输出图(格式为epoch_XXX.png)。

- 1SyMTRS: Benchmark Multi-Task Synthetic Dataset for Depth, Domain Adaptation and Super-Resolution in Aerial ImageryTOELT LLC AI实验室; 卢塞恩应用科学与艺术大学; 瑞士国防部·人工智能与模拟能力中心 · 2026年



