Ngoc Long Nguyen, Jérémy Anger, Axel Davy, Pablo Arias, Gabriele Facciolo
Centre Borelli, ENS Paris-Saclay
This repository is the official PyTorch implementation of L1BSR: Exploiting Detector Overlap for Self-Supervised SISR of Sentinel-2 L1B Imagery (Best Student Paper at EarthVision 2023).
L1BSR produces a 5m high-resolution (HR) output with all bands correctly registered from a single 10m low-resolution (LR) Sentinel-2 L1B image with misaligned bands. Note that L1BSR is trained on real data with self-supervision, i.e. without any ground truth HR targets.
There are two key modules integral to the training of the L1BSR:
- The REConstruction (REC) module: performs joint super-resolution and band-alignment for the L1B BGRN data.
- The Cross-Spectral Registration (CSR) module: produces a dense flow between 2 images of different spectral bands.
Both modules are trained with self-supervision. Note that the CSR is used only during the training of L1BSR, whereas at inference, only the REC is needed.
For your convenience we provide some test BGRN images (~10Mb) in /examples.
If you want a quick inspection of our two key modules REC and CSR, checkout our IPOL demo
We also provide the testing code main.py. Like in the demo, you can choose the task (super-resolution or cross-spectral registration) for our networks (REC or CSR, respectively) to perform.
Examples:
# Super-resolution: This code below super-resolves (x2) the image in "examples/00.tif"
# and saves it in "output.tif".
python main.py examples/00.tif output.tif --device cuda --task superresolution
# Cross-spectral registration: This code below aligns the bands Blue, Red, and NIR of
# the image in "examples/00.tif" to its Green band and saves the output in "output.tif".
python main.py examples/00.tif output.tif --device cuda --task registrationThe training codes for both the CSR and REC modules will be soon available. Stay tuned!
@inproceedings{nguyen2023l1bsr,
title={L1BSR: Exploiting Detector Overlap for Self-Supervised Single-Image Super-Resolution of Sentinel-2 L1B Imagery},
author={Nguyen, Ngoc Long and Anger, J{\'e}r{\'e}my and Davy, Axel and Arias, Pablo and Facciolo, Gabriele},
booktitle={Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition},
pages={2012--2022},
year={2023}
}
This project is released under the GPL-3.0 license. The codes are based on RCAN. Please also follow their licenses. Thanks for their awesome works.