Skip to content

Official PyTorch implementation of "PMAA: A Progressive Multi-scale Attention Autoencoder Model for High-Performance Cloud Removal from Multi-temporal Satellite Imagery" (ECAI 2023).

License

Notifications You must be signed in to change notification settings

XavierJiezou/PMAA

Repository files navigation

logo

PMAA: A Progressive Multi-scale Attention Autoencoder Model for High-Performance Cloud Removal from Multi-temporal Satellite Imagery

This repository is the official PyTorch implementation of the accepted paper PMAA of ECAI 2023.

Xuechao Zou1,*, Kai Li2,*, Junliang Xing2, Pin Tao1,2,†, Yachao Cui1

Qinghai University1Tsinghua University2

Paper Preprint | Project Page

pmaa

News

  • [2023/07/30] Code release.
  • [2023/07/16] PMAA got accepted by ECAI 2023.
  • [2023/03/29] PMAA is on arXiv now.

Requirements

To install dependencies:

pip install -r requirements.txt

To download datasets:

Training

To train the models in the paper, run these commands:

python train_old.py
python train_new.py

Evaluation

To evaluate my models on two datasets, run:

python test_old.py
python test_new.py

Pre-trained Models

You can download pretrained models here:

Results

res

Quantitative Results

exp

Qualitative Results

vis

Citation

If you use our code or models in your research, please cite with:

@article{zou2023pmaa,
  title={PMAA: A Progressive Multi-scale Attention Autoencoder Model for High-Performance Cloud Removal from Multi-temporal Satellite Imagery},
  author={Zou, Xuechao and Li, Kai and Xing, Junliang and Tao, Pin and Cui, Yachao},
  journal={European Conference on Artificial Intelligence (ECAI)},
  year={2023}
}

About

Official PyTorch implementation of "PMAA: A Progressive Multi-scale Attention Autoencoder Model for High-Performance Cloud Removal from Multi-temporal Satellite Imagery" (ECAI 2023).

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published