Skip to content

Code for TGRS paper "UCDFormer: Unsupervised Change Detection Using a Transformer-driven Image Translation"

Notifications You must be signed in to change notification settings

zhu-xlab/UCDFormer

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

31 Commits
 
 
 
 
 
 

Repository files navigation

UCDFormer

Change detection (CD) by comparing two bi-temporal images is a crucial task in remote sensing. With the advantages of requiring no cumbersome labeled change information, unsupervised CD has attracted extensive attention in the community. However, existing unsupervised CD approaches rarely consider the seasonal and style differences incurred by the illumination and atmospheric conditions in multi-temporal images. To this end, we propose a change detection with domain shift setting for remote sensing images. Furthermore, we present a novel unsupervised CD method using a light-weight transformer, called UCDFormer. Specifically, a transformer-driven image translation composed of a light-weight transformer and a domain-specific affinity weight is first proposed to mitigate domain shift between two images with real-time efficiency. After image translation, we can generate the difference map between the translated before-event image and the original after-event image. Then, a novel reliable pixel extraction module is proposed to select significantly changed/unchanged pixel positions by fusing the pseudo change maps of fuzzy c-means clustering and adaptive threshold. Finally, a binary change map is obtained based on these selected pixel pairs and a binary classifier. Experimental results on different unsupervised CD tasks with seasonal and style changes demonstrate the effectiveness of the proposed UCDFormer. For example, compared with several other related methods, UCDFormer improves performance on the Kappa coefficient by more than 12%. In addition, UCDFormer achieves excellent performance for earthquake-induced landslide detection when considering large-scale applications. Alt text Fig. 1. Change detection with domain shift, which is used to reflect style differences and seasonal differences between multi-temporal images.

Overview of UCDFormer

Alt text Fig. 2. Overview of UCDFormer. The architecture of the model is divided into three parts, i.e., image translation, reliable pixel extraction, and change map extraction. D, GN, CG, SBG represent downsampling, group normalization, channel group, and style bank generation, respectively.

Datasets

We evaluate the proposed network on three datasets: data with seasonal differences, data with style differences, and earthquake-induced landslide detection with style differences.

Pretrain Models

The Pretrain model of VGG is available at Link
The Pretrain model of the generator (StyleFormer) is available at Link

Run

After configuring the file path: run the code
python main_seasonal_change_from _summer_to_autumn.py for Seasonal Change from Summer to Autumn
python main_seasonal_change_from_spring to_winter.py for Seasonal Change from Spring to Winter
python main_style_changes.py for Style Changes
python main_landslide.py for Applications in Landslide Detection
Other applications just change the corresponding data path.

Citation

@ARTICLE{10251124, author={Xu, Qingsong and Shi, Yilei and Guo, Jianhua and Ouyang, Chaojun and Zhu, Xiao Xiang}, journal={IEEE Transactions on Geoscience and Remote Sensing}, title={UCDFormer: Unsupervised Change Detection Using a Transformer-Driven Image Translation}, year={2023}, volume={61}, number={}, pages={1-17}, doi={10.1109/TGRS.2023.3305334}}

About

Code for TGRS paper "UCDFormer: Unsupervised Change Detection Using a Transformer-driven Image Translation"

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published