The official implementation of **Gabor Feature Network for Transformer-based Building Change Detection Model in Remote Sensing** (ICIP 2024)
You can create an Anaconda environment named GabFormer
from environment.yml
.
conda env create -f environment.yml
conda activate GabFormer
Clone this repo:
git clone https://github.com/Ayana-Inria/GabFormer.git
cd GabFormer
- Change the dataset path in
data_config.py
. - Set the training parameter and model's hyperparameters in the training script
run_GabFormer.sh
in the folderscripts
. You can run the script file bysh scripts/run_GabFormer.sh
in the command environment.
You can find the evaluation script eval_GabFormer.sh
in the folder scripts
. You can run the script file by sh scripts/eval_GabFormer.sh
in the command environment.
"""
data structure
-dataroot
├─A
├─img1.png
...
├─B
├─img1.png
...
├─label
├─img1.png
...
└─list
├─val.txt
├─test.txt
└─train.txt
# In list/ folder, prepare text files of the splits and list down all filenames of each split
# for example:
list/train.txt
img1.png
img32.png
...
list/test.txt
img2.png
img15.png
...
list/val.txt
img54.png
img100.png
...
"""
A
: pre-change images;
B
: post-change images;
label
: binary labels;
list
: contains train.txt, val.txt and test.txt
, each file records the image names (XXX.png) in the change detection dataset.
GabFormer was tested on two datasets for remote sensing building change detection.
-
LEVIR-CD
-
WHU-CD
We utilized the same cropping method as what is provided here to the size of 256 x 256.
The code is released under the GPL-3.0-only license. See LICENSE
file for more details.
If you use this code for your research, please cite our paper (to be updated):
@inproceedings{GabFormer,
title={Gabor Feature Network for Transformer-based Building Change Detection Model in Remote Sensing},
author={Priscilla Indira Osa and Josiane Zerubia and Zoltan Kato},
year={2024},
}
Our GabFormer is implemented based on the code provided in repositories below: