Awesome
Semantic-guidance-and-spatial-localization-network
Introduction
This repo is the official implementation of "Exchanging Dual-Encoder–Decoder: A New Strategy for Change Detection With Semantic Guidance and Spatial Localization"
Install dependencies
- Install CUDA
- Install Pytorch 1.12 or later
- Install dependencies
Use the following code in command line to install dependencies.
pip install -r requirements.txt
Data
Using any change detection dataset you want, but organize dataset path as follows. dataset_name
is name of change detection dataset, you can set whatever you want.
dataset_name
├─train
│ ├─label
│ ├─t1
│ └─t2
├─val
│ ├─label
│ ├─t1
│ └─t2
└─test
├─label
├─t1
└─t2
Below are some binary change detection dataset you may want.
Paper: Fully convolutional networks for multisource building extraction from an open aerial and satellite imagery data set
Paper: A deeply supervised image fusion network for change detection in high resolution bi-temporal remote sensing images
Paper: A Spatial-Temporal Attention-Based Method and a New Dataset for Remote Sensing Image Change Detection
Paper: SemiCDNet: A Semisupervised Convolutional Neural Network for Change Detection in High Resolution Remote-Sensing Images
Paper: SYSU-CD: A new change detection dataset in "A Deeply-supervised Attention Metric-based Network and an Open Aerial Image Dataset for Remote Sensing Change Detection"
Paper: CHANGE DETECTION IN REMOTE SENSING IMAGES USING CONDITIONAL ADVERSARIAL NETWORKS
Paper: Semantic feature-constrained multitask siamese network for building change detection in high-spatial-resolution remote sensing imagery
Paper: S2Looking: A Satellite Side-Looking Dataset for Building Change Detection
Start
For training, run the following code in command line.
python train.py
If you want to debug while training, run the following code in command line.
python -m ipdb train.py
For test and inference, run the following code in command line.
python inference.py
Config
All the configs of dataset, training, validation and test are put in the file "utils/path_hyperparameter.py", you can change the configs in this file.
Citation
If you use this work in your research, please cite:
@article{zhao2023exchanging, title={Exchanging Dual-Encoder--Decoder: A New Strategy for Change Detection With Semantic Guidance and Spatial Localization}, author={Zhao, Sijie and Zhang, Xueliang and Xiao, Pengfeng and He, Guangjun}, journal={IEEE Transactions on Geoscience and Remote Sensing}, volume={61}, pages={1--16}, year={2023}, publisher={IEEE} }
简介
这个项目是"Exchanging Dual-Encoder–Decoder: A New Strategy for Change Detection With Semantic Guidance and Spatial Localization"的官方pytorch实现
下载需要的库
- 下载CUDA
- 下载1.12或者更新的pytorch
- 下载其他需要的包
在命令行中运行下面的命令下载其他需要的包
pip install -r requirements.txt
数据
你可以使用任何你想使用的变化检测数据集,但是文件组织方式需要按照下面的来。dataset_name
是你设置的变化检测数据集的名字。
dataset_name
├─train
│ ├─label
│ ├─t1
│ └─t2
├─val
│ ├─label
│ ├─t1
│ └─t2
└─test
├─label
├─t1
└─t2
下面是一些你可能需要的二分类变化检测数据集。
Paper: Fully convolutional networks for multisource building extraction from an open aerial and satellite imagery data set
Paper: A deeply supervised image fusion network for change detection in high resolution bi-temporal remote sensing images
Paper: A Spatial-Temporal Attention-Based Method and a New Dataset for Remote Sensing Image Change Detection
Paper: SemiCDNet: A Semisupervised Convolutional Neural Network for Change Detection in High Resolution Remote-Sensing Images
Paper: SYSU-CD: A new change detection dataset in "A Deeply-supervised Attention Metric-based Network and an Open Aerial Image Dataset for Remote Sensing Change Detection"
Paper: CHANGE DETECTION IN REMOTE SENSING IMAGES USING CONDITIONAL ADVERSARIAL NETWORKS
Paper: Semantic feature-constrained multitask siamese network for building change detection in high-spatial-resolution remote sensing imagery
Paper: S2Looking: A Satellite Side-Looking Dataset for Building Change Detection
开始
在命令行中运行下面的代码来开始训练
python train.py
如果你想在训练的时候进行调试,在命令行中运行下面的命令
python -m ipdb train.py
在命令行中运行下面的代码来开始测试或者推理
python inference.py
设置
所有和数据集、训练、验证和测试的设置都放在了“utils/path_hyperparameter.py”文件中,你可以在这个文件里修改设置
引用
如果你在你的研究中用到了这篇工作的内容,请引用:
@article{zhao2023exchanging, title={Exchanging Dual-Encoder--Decoder: A New Strategy for Change Detection With Semantic Guidance and Spatial Localization}, author={Zhao, Sijie and Zhang, Xueliang and Xiao, Pengfeng and He, Guangjun}, journal={IEEE Transactions on Geoscience and Remote Sensing}, volume={61}, pages={1--16}, year={2023}, publisher={IEEE} }