下载资源后端资源详情
AttentionGAN-for-Cloud-removal-master.zip
大小:12.58MB
价格:18积分
下载量:0
评分:
5.0
上传者:qq_69854365
更新日期:2024-08-13

基于注意机制的GAN用于SAR和光学图像去云

资源文件列表(大概)

文件名
大小
AttentionGAN-for-Cloud-removal-master/
-
AttentionGAN-for-Cloud-removal-master/README.md
2.23KB
AttentionGAN-for-Cloud-removal-master/data/
-
AttentionGAN-for-Cloud-removal-master/data/__init__.py
-
AttentionGAN-for-Cloud-removal-master/data/__pycache__/
-
AttentionGAN-for-Cloud-removal-master/data/__pycache__/__init__.cpython-37.pyc
149B
AttentionGAN-for-Cloud-removal-master/data/__pycache__/__init__.cpython-38.pyc
145B
AttentionGAN-for-Cloud-removal-master/data/__pycache__/aligned_dataset.cpython-37.pyc
2.45KB
AttentionGAN-for-Cloud-removal-master/data/__pycache__/base_data_loader.cpython-37.pyc
690B
AttentionGAN-for-Cloud-removal-master/data/__pycache__/base_data_loader.cpython-38.pyc
706B
AttentionGAN-for-Cloud-removal-master/data/__pycache__/base_dataset.cpython-37.pyc
2.52KB
AttentionGAN-for-Cloud-removal-master/data/__pycache__/base_dataset.cpython-38.pyc
2.55KB
AttentionGAN-for-Cloud-removal-master/data/__pycache__/custom_dataset_data_loader.cpython-37.pyc
2.1KB
AttentionGAN-for-Cloud-removal-master/data/__pycache__/custom_dataset_data_loader.cpython-38.pyc
2.14KB
AttentionGAN-for-Cloud-removal-master/data/__pycache__/data_loader.cpython-37.pyc
414B
AttentionGAN-for-Cloud-removal-master/data/__pycache__/data_loader.cpython-38.pyc
414B
AttentionGAN-for-Cloud-removal-master/data/__pycache__/image_folder.cpython-37.pyc
2.03KB
AttentionGAN-for-Cloud-removal-master/data/__pycache__/image_folder.cpython-38.pyc
2.07KB
AttentionGAN-for-Cloud-removal-master/data/__pycache__/unaligned_dataset.cpython-37.pyc
1.99KB
AttentionGAN-for-Cloud-removal-master/data/__pycache__/unaligned_dataset.cpython-38.pyc
2KB
AttentionGAN-for-Cloud-removal-master/data/__pycache__/unaligned_dataset3.cpython-37.pyc
2.04KB
AttentionGAN-for-Cloud-removal-master/data/__pycache__/unaligned_dataset_sar.cpython-37.pyc
2.2KB
AttentionGAN-for-Cloud-removal-master/data/__pycache__/unaligned_dataset_sar.cpython-38.pyc
2.22KB
AttentionGAN-for-Cloud-removal-master/data/aligned_dataset.py
2.35KB
AttentionGAN-for-Cloud-removal-master/data/aligned_random_dataset.py
1.47KB
AttentionGAN-for-Cloud-removal-master/data/base_data_loader.py
195B
AttentionGAN-for-Cloud-removal-master/data/base_dataset.py
2.43KB
AttentionGAN-for-Cloud-removal-master/data/base_dataset.pyc
3.63KB
AttentionGAN-for-Cloud-removal-master/data/coco_dataset.py
2.13KB
AttentionGAN-for-Cloud-removal-master/data/cocoseg_dataset.py
3.82KB
AttentionGAN-for-Cloud-removal-master/data/custom_dataset_data_loader.py
1.83KB
AttentionGAN-for-Cloud-removal-master/data/data_loader.py
229B
AttentionGAN-for-Cloud-removal-master/data/data_loader.pyc
539B
AttentionGAN-for-Cloud-removal-master/data/image_folder.py
1.9KB
AttentionGAN-for-Cloud-removal-master/data/image_folder.pyc
2.95KB
AttentionGAN-for-Cloud-removal-master/data/single_dataset.py
1.07KB
AttentionGAN-for-Cloud-removal-master/data/unaligned_dataset.py
1.94KB
AttentionGAN-for-Cloud-removal-master/data/unaligned_dataset3.py
2.04KB
AttentionGAN-for-Cloud-removal-master/data/unaligned_dataset_sar.py
2.31KB
AttentionGAN-for-Cloud-removal-master/environment.yaml
2.32KB
AttentionGAN-for-Cloud-removal-master/fig/
-
AttentionGAN-for-Cloud-removal-master/fig/pipline.png
5.53MB
AttentionGAN-for-Cloud-removal-master/fig/result.png
6.61MB
AttentionGAN-for-Cloud-removal-master/map.npy
256.13KB
AttentionGAN-for-Cloud-removal-master/metrics.py
1.15KB
AttentionGAN-for-Cloud-removal-master/models/
-
AttentionGAN-for-Cloud-removal-master/models/Saroptgan.py
6.01KB
AttentionGAN-for-Cloud-removal-master/models/__init__.py
-
AttentionGAN-for-Cloud-removal-master/models/__pycache__/
-
AttentionGAN-for-Cloud-removal-master/models/__pycache__/Saroptgan.cpython-37.pyc
4.91KB
AttentionGAN-for-Cloud-removal-master/models/__pycache__/__init__.cpython-37.pyc
151B
AttentionGAN-for-Cloud-removal-master/models/__pycache__/__init__.cpython-38.pyc
147B
AttentionGAN-for-Cloud-removal-master/models/__pycache__/base_model.cpython-37.pyc
3.13KB
AttentionGAN-for-Cloud-removal-master/models/__pycache__/base_model.cpython-38.pyc
3.19KB
AttentionGAN-for-Cloud-removal-master/models/__pycache__/cycle_attn_gan_model.cpython-37.pyc
8.51KB
AttentionGAN-for-Cloud-removal-master/models/__pycache__/cycle_attn_gan_model_sar.cpython-37.pyc
9.14KB
AttentionGAN-for-Cloud-removal-master/models/__pycache__/cycle_attn_gan_model_sar2.cpython-37.pyc
8.92KB
AttentionGAN-for-Cloud-removal-master/models/__pycache__/cycle_gan_model.cpython-37.pyc
6.47KB
AttentionGAN-for-Cloud-removal-master/models/__pycache__/models.cpython-37.pyc
2.11KB
AttentionGAN-for-Cloud-removal-master/models/__pycache__/models.cpython-38.pyc
2.13KB
AttentionGAN-for-Cloud-removal-master/models/__pycache__/networks.cpython-37.pyc
29.22KB
AttentionGAN-for-Cloud-removal-master/models/__pycache__/networks.cpython-38.pyc
28.52KB
AttentionGAN-for-Cloud-removal-master/models/__pycache__/pix2pix_attn_model.cpython-37.pyc
6.65KB
AttentionGAN-for-Cloud-removal-master/models/__pycache__/pix2pix_attn_model.cpython-38.pyc
7.46KB
AttentionGAN-for-Cloud-removal-master/models/__pycache__/pix2pix_attn_model2.cpython-37.pyc
5.52KB
AttentionGAN-for-Cloud-removal-master/models/__pycache__/pix2pix_attn_model3.cpython-37.pyc
6.79KB
AttentionGAN-for-Cloud-removal-master/models/__pycache__/pix2pix_attn_model31.cpython-37.pyc
6.46KB
AttentionGAN-for-Cloud-removal-master/models/__pycache__/pix2pix_attn_model4.cpython-37.pyc
7.4KB
AttentionGAN-for-Cloud-removal-master/models/__pycache__/pix2pix_attn_model4.cpython-38.pyc
7.48KB
AttentionGAN-for-Cloud-removal-master/models/__pycache__/pix2pix_attn_model5.cpython-37.pyc
6.57KB
AttentionGAN-for-Cloud-removal-master/models/__pycache__/pix2pix_attn_model6.cpython-37.pyc
6.65KB
AttentionGAN-for-Cloud-removal-master/models/__pycache__/pix2pix_attn_model7.cpython-37.pyc
7.46KB
AttentionGAN-for-Cloud-removal-master/models/__pycache__/pix2pix_attn_model8.cpython-37.pyc
6.57KB
AttentionGAN-for-Cloud-removal-master/models/__pycache__/pix2pix_model.cpython-37.pyc
5.09KB
AttentionGAN-for-Cloud-removal-master/models/__pycache__/pix2pix_model2.cpython-37.pyc
5.1KB
AttentionGAN-for-Cloud-removal-master/models/__pycache__/pix2pix_model3.cpython-37.pyc
4.65KB
AttentionGAN-for-Cloud-removal-master/models/base_model.py
2.36KB
AttentionGAN-for-Cloud-removal-master/models/base_model.pyc
4.1KB
AttentionGAN-for-Cloud-removal-master/models/cycle_attn_gan_model.py
13.46KB
AttentionGAN-for-Cloud-removal-master/models/cycle_attn_gan_model.pyc
12.89KB
AttentionGAN-for-Cloud-removal-master/models/cycle_attn_gan_model_sar.py
14.72KB
AttentionGAN-for-Cloud-removal-master/models/cycle_attn_gan_model_sar2.py
14.1KB
AttentionGAN-for-Cloud-removal-master/models/cycle_gan_model.py
9.17KB
AttentionGAN-for-Cloud-removal-master/models/map.npy
256.13KB
AttentionGAN-for-Cloud-removal-master/models/models.py
2.5KB
AttentionGAN-for-Cloud-removal-master/models/models.pyc
4.13KB
AttentionGAN-for-Cloud-removal-master/models/networks.py
41.97KB
AttentionGAN-for-Cloud-removal-master/models/pix2pix_attn_model.py
10.92KB
AttentionGAN-for-Cloud-removal-master/models/pix2pix_model.py
6.09KB
AttentionGAN-for-Cloud-removal-master/models/pix2pix_model2.py
6.26KB
AttentionGAN-for-Cloud-removal-master/models/pix2pix_model3.py
5.47KB
AttentionGAN-for-Cloud-removal-master/models/test_model.py
1.58KB
AttentionGAN-for-Cloud-removal-master/options/
-
AttentionGAN-for-Cloud-removal-master/options/__pycache__/
-
AttentionGAN-for-Cloud-removal-master/options/__pycache__/base_options.cpython-37.pyc
5.11KB
AttentionGAN-for-Cloud-removal-master/options/__pycache__/base_options.cpython-38.pyc
5.13KB
AttentionGAN-for-Cloud-removal-master/options/__pycache__/test_options.cpython-37.pyc
994B
AttentionGAN-for-Cloud-removal-master/options/__pycache__/test_options.cpython-38.pyc
1KB
AttentionGAN-for-Cloud-removal-master/options/__pycache__/train_options.cpython-37.pyc
2.79KB
AttentionGAN-for-Cloud-removal-master/options/__pycache__/train_options.cpython-38.pyc
2.79KB
AttentionGAN-for-Cloud-removal-master/options/base_options.py
7.09KB
AttentionGAN-for-Cloud-removal-master/options/test_options.py
846B
AttentionGAN-for-Cloud-removal-master/options/train_options.py
3.26KB
AttentionGAN-for-Cloud-removal-master/options/train_options.pyc
4.08KB
AttentionGAN-for-Cloud-removal-master/pytorch_ssim/
-
AttentionGAN-for-Cloud-removal-master/pytorch_ssim/__init__.py
2.57KB
AttentionGAN-for-Cloud-removal-master/pytorch_ssim/__pycache__/
-
AttentionGAN-for-Cloud-removal-master/pytorch_ssim/__pycache__/__init__.cpython-37.pyc
2.53KB
AttentionGAN-for-Cloud-removal-master/pytorch_ssim/__pycache__/__init__.cpython-38.pyc
2.57KB
AttentionGAN-for-Cloud-removal-master/requirements.txt
1.83KB
AttentionGAN-for-Cloud-removal-master/scripts/
-
AttentionGAN-for-Cloud-removal-master/scripts/test.sh
222B
AttentionGAN-for-Cloud-removal-master/scripts/train.sh
319B
AttentionGAN-for-Cloud-removal-master/test.py
1.29KB
AttentionGAN-for-Cloud-removal-master/train.py
1.88KB
AttentionGAN-for-Cloud-removal-master/util/
-
AttentionGAN-for-Cloud-removal-master/util/__pycache__/
-
AttentionGAN-for-Cloud-removal-master/util/__pycache__/html.cpython-37.pyc
2.22KB
AttentionGAN-for-Cloud-removal-master/util/__pycache__/html.cpython-38.pyc
2.24KB
AttentionGAN-for-Cloud-removal-master/util/__pycache__/image_pool.cpython-37.pyc
1.71KB
AttentionGAN-for-Cloud-removal-master/util/__pycache__/image_pool.cpython-38.pyc
1.7KB
AttentionGAN-for-Cloud-removal-master/util/__pycache__/util.cpython-37.pyc
4.08KB
AttentionGAN-for-Cloud-removal-master/util/__pycache__/util.cpython-38.pyc
4.15KB
AttentionGAN-for-Cloud-removal-master/util/__pycache__/visualizer.cpython-37.pyc
4.67KB
AttentionGAN-for-Cloud-removal-master/util/__pycache__/visualizer.cpython-38.pyc
4.71KB
AttentionGAN-for-Cloud-removal-master/util/get_data.py
3.43KB
AttentionGAN-for-Cloud-removal-master/util/html.py
1.82KB
AttentionGAN-for-Cloud-removal-master/util/html.pyc
3.03KB
AttentionGAN-for-Cloud-removal-master/util/image_pool.py
2.45KB
AttentionGAN-for-Cloud-removal-master/util/image_pool.pyc
2.32KB
AttentionGAN-for-Cloud-removal-master/util/inception_classify.py
2.92KB
AttentionGAN-for-Cloud-removal-master/util/png.py
978B
AttentionGAN-for-Cloud-removal-master/util/util.py
3.19KB
AttentionGAN-for-Cloud-removal-master/util/util.pyc
5.72KB
AttentionGAN-for-Cloud-removal-master/util/visualizer.py
6.34KB
AttentionGAN-for-Cloud-removal-master/util/visualizer.pyc
6.36KB

资源内容介绍

这个库为我们的论文“通过基于注意机制的GAN使用SAR和光学图像去云”提供PyTorch代码。此代码基于的实现注意GAN和CycleGAN.
# Cloud removal using SAR and optical images via attention mechanism-based GANThis repository provides the PyTorch code for our paper “Cloud removal using SAR and optical images via attention mechanism-based GAN”. This code is based on the implementation of [Attention-GAN](https://github.com/xinyuanc91/Attention-GAN) and [CycleGAN](https://github.com/junyanz/pytorch-CycleGAN-and-pix2pix). ## OverviewThe pipline of our method:![图片](/fig/pipline.png "method pipline")The result of our method:![图片](/fig/result.png "result")*Cloud removal results in the real data experiment. Each row corresponds to an example of the results. Column (a) shows the SAR images. Column (b) shows the attention map. Column (c) shows the cloud images. Columns (d) to (h) correspond to the results obtained by the pix2pix model, the SAR-opt-GAN model, the Simulation-Fusion GAN model, the GLF-CR model, and the proposed model. Column (i) shows the ground truth images.*## DataWe use part of the data from the [SEN12MS-CR dataset](https://patricktum.github.io/cloud_removal/sen12mscr/) in the paper.You should organize your data into a format like this, replacing them with the data directory in this code:```data│└───train/test│ ││ └───trainA/testA #cloud images│ │ │ 1.png│ │ │ 2.png│ │ │ ...│ ││ └───trainB/testB #cloudless images│ │ │ 1.png│ │ │ 2.png│ │ │ ...│ ││ └───trainC/testC #SAR images│ │ │ 1.png│ │ │ 2.png│ │ │ ...│ ```## EnvironmentYou can Install dependencies via:```bashconda env create --file environment.yml```## Train sh scripts/train.sh## Test sh scripts/test.sh## CitationIf you find this repository/work helpful in your research, welcome to cite the paper.```@article{zhang2023cloud, title={Cloud removal using SAR and optical images via attention mechanism-based GAN}, author={Zhang, Shuai and Li, Xiaodi and Zhou, Xingyu and Wang, Yuning and Hu, Yue}, journal={Pattern Recognition Letters}, volume={175}, pages={8--15}, year={2023}, publisher={Elsevier}}```

用户评论 (0)

发表评论

captcha