Semantic Guided Large Scale Factor Remote Sensing Image Super-resolution with Generative Diffusion Prior
Officical code for Paper
# clone this repo
git clone https://github.com/wwangcece/SGDM.git
# create an environment with python >= 3.9
conda create -n SGDM python=3.9
conda activate SGDM
pip install -r requirements.txt
Download the model and place it in the checkpoints/ folder
For copyright reasons, we can only provide the geographic sampling points in the data and the download scripts of the remote sensing images. To download vector maps, you need to register a maptiler account and subscribe to the package.
First please modify the validation data set configuration files at configs/dataset
python inference_refsr_batch_simu.py \
--ckpt checkpoints/SGDM-syn.ckpt \
--config configs/model/refsr_simu.yaml \
--val_config configs/dataset/reference_sr_val_simu.yaml \
--output path/to/your/outpath \
--steps 50 \
--device cuda:0 \
For style sampling
python inference_refsr_batch_real.py \
--ckpt checkpoints/SGDM-real.ckpt \
--config configs/model/refsr_real.yaml \
--val_config configs/dataset/reference_sr_val_real.yaml \
--sample_style true \
--ckpt_flow_mean checkpoints/flow_tanh_mini_mean \
--ckpt_flow_std checkpoints/flow_tanh_mini_std \
--output path/to/your/outpath \
--steps 50 \
--device cuda:0 \
For style guidance
python inference_refsr_batch_real.py \
--ckpt checkpoints/SGDM-real.ckpt \
--config configs/model/refsr_real.yaml \
--val_config configs/dataset/reference_sr_val_real.yaml \
--output 50 path/to/your/outpath \
--steps 50 \
--device cuda:0 \
TBD
Please cite us if our work is useful for your research.
@article{wang2024semantic,
title={Semantic Guided Large Scale Factor Remote Sensing Image Super-resolution with Generative Diffusion Prior},
author={Wang, Ce and Sun, Wanjie},
journal={arXiv preprint arXiv:2405.07044},
year={2024}
}
This project is based on Diffbir. Thanks for their awesome work.
If you have any questions, please feel free to contact with me at [email protected].