AFENet_MCD

Adversarial Feature Equilibrium Network for Multimodal Change Detection in Heterogeneous Remote Sensing Images

This is a PyTorch/GPU implementation of the paper Adversarial Feature Equilibrium Network for Multimodal Change Detection in Heterogeneous Remote Sensing Images:

@ARTICLE{pu2024adversarial,
  author={Pu, Yan and Gong, Maoguo and Liu, Tongfei and Zhang, Mingyang and Gao, Tianqi and Jiang, Fenlong and Hu, Xiaobo},
  journal={IEEE Transactions on Geoscience and Remote Sensing}, 
  title={Adversarial Feature Equilibrium Network for Multimodal Change Detection in Heterogeneous Remote Sensing Images}, 
  year={2024},
  volume={62},
  number={},
  pages={1-17},

Requirements

• python 3.7.16
• torch 1.13.1
• torchvision 0.14.1
• opencv 4.7.0.72
• numpy 1.21.5
• pandas 1.3.5

Dataset

Multimodal Dataset	Download Link	Source
MT-Wuhan	Download	Paper
MT-HTCD	Download	Paper

Installation

For convenience, our pre-trained models can be downloaded directly here:

Model	Dataset	F1-Score	Params
G_model	MT-Wuhan	60.03	297.8M
D_model	MT-Wuhan	60.03	11.7M
G_model	MT-HTCD	97.15	297.8M
D_model	MT-HTCD	97.15	11.7M

Usage

Train and Test

  python main.py