This repository contains the code of the paper Trustworthy Super-Resolution of Multispectral Sentinel-2 Imagery with Latent Diffusion.
Run this model in Google Colab! The notebook now supports the unrestricted selection of your own coordinates and automated download and SR of S2 imagery.
PLEASE NOTE:
- This has left the experimental stage with v1.0.0
- If you are interested in applying SR to the 10m and 20m bands, please check out SEN2SR. Use SEN2SR and LDSR-S2 here:
If you use this model in your work, please cite
@ARTICLE{ldsrs2,
author={Donike, Simon and Aybar, Cesar and Gómez-Chova, Luis and Kalaitzis, Freddie},
journal={IEEE Journal of Selected Topics in Applied Earth Observations and Remote Sensing},
title={Trustworthy Super-Resolution of Multispectral Sentinel-2 Imagery With Latent Diffusion},
year={2025},
volume={18},
number={},
pages={6940-6952},
doi={10.1109/JSTARS.2025.3542220}}
pip install opensr-model
Minimal Example
import opensr_model # import pachage
model = opensr_model.SRLatentDiffusion(config, device=device) # create model
model.load_pretrained(config.ckpt_version) # load checkpoint
sr = model.forward(torch.rand(1,4,128,128), custom_steps=100) # run SR
Run the 'demo.py' file to gain an understanding how the package works. It will SR and example tensor and save the according uncertainty map.
Output of demo.py file:
The model should load automatically with the model.load_pretrained command. Alternatively, the checkpoints can be found on HuggingFace
This package contains the latent-diffusion model to super-resolute 10 and 20m bands of Sentinel-2. This repository contains the bare model. It can be embedded in the "opensr-utils" package in order to be applied to Sentinel-2 Imagery.
Examples on S2NAIP training dataset
This repository has left the experimental stage with the publication of v1.0.0.
Some example Sr scenes can be found as super-resoluted tiffs on Doogle Drive. Scenes available:
- Buenos Aires, Argentina
- Blue Mountains, Australia
- Louisville, USA
- Kutahya, Türkyie
- Catalunya, Spain