RealLQ250
收藏DreamClear 数据集概述
数据集简介
DreamClear 是一个专注于高容量真实世界图像恢复的数据集,旨在通过隐私安全的数据集构建方法提升图像恢复的质量。
主要贡献者
- Yuang Ai<sup>1,2</sup>
- Xiaoqiang Zhou<sup>1,4</sup>
- Huaibo Huang<sup>1,2</sup>
- Xiaotian Han<sup>3</sup>
- Zhengyu Chen<sup>3</sup>
- Quanzeng You<sup>3</sup>
- Hongxia Yang<sup>3</sup>
机构
- <sup>1</sup>MAIS & NLPR, Institute of Automation, Chinese Academy of Sciences
- <sup>2</sup>School of Artificial Intelligence, University of Chinese Academy of Sciences
- <sup>3</sup>ByteDance, Inc
- <sup>4</sup>University of Science and Technology of China
数据集发布信息
- 2024.10.25: 发布
RealLQ250基准测试集,包含 250 张真实世界低质量图像。 - 2024.10.25: 发布训练和推理代码(256->1024)以及预训练模型。
- 2024.10.24: 创建此仓库。
数据集内容
- RealLQ250: 包含 250 张真实世界低质量图像的基准测试集。
数据集下载
- 预训练模型和
RealLQ250基准测试集可从 Huggingface 和 Google Drive 下载。
数据集使用
训练数据准备
-
生成配对数据: shell python3 tools/make_paired_data.py --gt_path gt_path1 gt_path2 ...
--save_dir /path/to/save/folder/ --epoch 1 -
提取 T5 特征: shell python3 tools/extract_t5_features.py --t5_ckpt /path/to/t5-v1_1-xxl --caption_folder /path/to/caption/folder --save_npz_folder /path/to/save/npz/folder
训练 DreamClear
shell python3 -m torch.distributed.launch --nproc_per_node=8 --nnodes=... --node_rank=... --master_addr=... --master_port=... train_dreamclear.py configs/DreamClear/DreamClear_Train.py --load_from /path/to/PixArt-XL-2-1024-MS.pth --vae_pretrained /path/to/sd-vae-ft-ema --swinir_pretrained /path/to/general_swinir_v1.ckpt --val_image /path/to/RealLQ250/lq/val_image.png --val_npz /path/to/RealLQ250/npz/val_image.npz --work_dir experiments/train_dreamclear
推理
shell python3 -m torch.distributed.launch --nproc_per_node 1 --master_port 1234 test_1024.py configs/DreamClear/DreamClear_Test.py --dreamclear_ckpt /path/to/DreamClear-1024.pth --swinir_ckpt /path/to/general_swinir_v1.ckpt --vae_ckpt /path/to/sd-vae-ft-ema --lre --cfg_scale 4.5 --color_align wavelet --image_path /path/to/RealLQ250/lq --npz_path /path/to/RealLQ250/npz --save_dir validation
许可证
本数据集和代码基于 Apache 2.0 许可证。
引用
@article{ai2024dreamclear, title={DreamClear: High-Capacity Real-World Image Restoration with Privacy-Safe Dataset Curation}, author={Ai, Yuang and Zhou, Xiaoqiang and Huang, Huaibo and Han, Xiaotian and Chen, Zhengyu and You, Quanzeng and Yang, Hongxia}, journal={Advances in Neural Information Processing Systems}, year={2024} }




