CAGNet

The pytorch implementation for CAGNet in paper "Cross-Attention Guided Group Aggregation Network for Cropland Change Detection".

Requirements

• Python 3.6
• Pytorch 1.7.0

Datasets Preparation

The path list in the datasest folder is as follows:

|—train

• ||—A

• ||—B

• ||—OUT

|—val

• ||—A

• ||—B

• ||—OUT

|—test

• ||—A

• ||—B

• ||—OUT

where A contains pre-temporal images, B contains post-temporal images, and OUT contains ground truth images.

Train

• python train.py --dataset-dir dataset-path

Test

• python eval.py --ckp-paths weight-path --dataset-dir dataset-path

Visualization

• python visualization visualization.py --ckp-paths weight-path --dataset-dir dataset-path (Note that batch-size must be 1 when using visualization.py)
• Besides, you can adjust the parameter of full_to_color to change the color