Upload 6 files

configs/_base_/CLCD_config.py

@@ -0,0 +1,45 @@
+dataset_config = dict(
+    type = 'CLCD',
+    data_root = 'data/CLCD',
+    train_mode = dict(
+        transform = dict(
+            # RandomSizeAndCrop = {"size": 512, "crop_nopad": False},
+            RandomHorizontallyFlip = {"p": 0.5},
+            RandomVerticalFlip = {"p": 0.5},
+            RandomGaussianBlur = None,
+        ),
+        loader = dict(
+            batch_size = 4,
+            num_workers = 4,
+            pin_memory=True,
+            shuffle = True,
+            drop_last = True
+        ),
+    ),
+    
+    val_mode = dict(
+        transform = dict(
+            # RandomSizeAndCrop = {"size": 512, "crop_nopad": False},
+        ),
+        loader = dict(
+            batch_size = 4,
+            num_workers = 4,
+            pin_memory=True,
+            shuffle = False,
+            drop_last = False
+        )
+    ),
+
+    test_mode = dict(
+        transform = dict(
+            # RandomSizeAndCrop = {"size": 512, "crop_nopad": False},
+        ),
+        loader = dict(
+            batch_size = 4,
+            num_workers = 4,
+            pin_memory=True,
+            shuffle = False,
+            drop_last = False
+        )
+    ),
+)

configs/_base_/DSIFN_config.py

@@ -0,0 +1,45 @@
+dataset_config = dict(
+    type = 'DSIFN',
+    data_root = 'data/DSIFN',
+    train_mode = dict(
+        transform = dict(
+            # RandomSizeAndCrop = {"size": 512, "crop_nopad": False},
+            RandomHorizontallyFlip = {"p": 0.5},
+            RandomVerticalFlip = {"p": 0.5},
+            RandomGaussianBlur = None,
+        ),
+        loader = dict(
+            batch_size = 4,
+            num_workers = 4,
+            pin_memory=True,
+            shuffle = True,
+            drop_last = True
+        ),
+    ),
+    
+    val_mode = dict(
+        transform = dict(
+            # RandomSizeAndCrop = {"size": 512, "crop_nopad": False},
+        ),
+        loader = dict(
+            batch_size = 4,
+            num_workers = 4,
+            pin_memory=True,
+            shuffle = False,
+            drop_last = False
+        )
+    ),
+
+    test_mode = dict(
+        transform = dict(
+            # RandomSizeAndCrop = {"size": 512, "crop_nopad": False},
+        ),
+        loader = dict(
+            batch_size = 4,
+            num_workers = 4,
+            pin_memory=True,
+            shuffle = False,
+            drop_last = False
+        )
+    ),
+)

configs/_base_/LEVIRCD_config.py

@@ -0,0 +1,45 @@
+dataset_config = dict(
+    type = 'LEVIRCD',
+    data_root = 'data/LEVIR_CD',
+    train_mode = dict(
+        transform = dict(
+            RandomSizeAndCrop = {"size": 256, "crop_nopad": False},
+            RandomHorizontallyFlip = {"p": 0.5},
+            RandomVerticalFlip = {"p": 0.5},
+            RandomGaussianBlur = None,
+        ),
+        loader = dict(
+            batch_size = 8,
+            num_workers = 4,
+            pin_memory=True,
+            shuffle = True,
+            drop_last = True
+        ),
+    ),
+    
+    val_mode = dict(
+        transform = dict(
+            # RandomSizeAndCrop = {"size": 256, "crop_nopad": False},
+        ),
+        loader = dict(
+            batch_size = 8,
+            num_workers = 4,
+            pin_memory=True,
+            shuffle = False,
+            drop_last = False
+        )
+    ),
+
+    test_mode = dict(
+        transform = dict(
+            # RandomSizeAndCrop = {"size": 256, "crop_nopad": False},
+        ),
+        loader = dict(
+            batch_size = 8,
+            num_workers = 4,
+            pin_memory=True,
+            shuffle = False,
+            drop_last = False
+        )
+    ),
+)

configs/_base_/SYSUCD_config.py

@@ -0,0 +1,45 @@
+dataset_config = dict(
+    type = 'SYSUCD',
+    data_root = 'data/SYSU_CD',
+    train_mode = dict(
+        transform = dict(
+            # RandomSizeAndCrop = {"size": 256, "crop_nopad": False},
+            RandomHorizontallyFlip = {"p": 0.5},
+            RandomVerticalFlip = {"p": 0.5},
+            RandomGaussianBlur = None,
+        ),
+        loader = dict(
+            batch_size = 16,
+            num_workers = 4,
+            pin_memory=True,
+            shuffle = True,
+            drop_last = True
+        ),
+    ),
+    
+    val_mode = dict(
+        transform = dict(
+            # RandomSizeAndCrop = {"size": 256, "crop_nopad": False},
+        ),
+        loader = dict(
+            batch_size = 16,
+            num_workers = 4,
+            pin_memory=True,
+            shuffle = False,
+            drop_last = False
+        )
+    ),
+
+    test_mode = dict(
+        transform = dict(
+            # RandomSizeAndCrop = {"size": 256, "crop_nopad": False},
+        ),
+        loader = dict(
+            batch_size = 16,
+            num_workers = 4,
+            pin_memory=True,
+            shuffle = False,
+            drop_last = False
+        )
+    ),
+)

configs/_base_/WHUCD_config.py

@@ -0,0 +1,45 @@
+dataset_config = dict(
+    type = 'WHUCD',
+    data_root = 'data/WHU_CD',
+    train_mode = dict(
+        transform = dict(
+            # RandomSizeAndCrop = {"size": 256, "crop_nopad": False},
+            RandomHorizontallyFlip = {"p": 0.5},
+            RandomVerticalFlip = {"p": 0.5},
+            RandomGaussianBlur = None,
+        ),
+        loader = dict(
+            batch_size = 8,
+            num_workers = 4,
+            pin_memory=True,
+            shuffle = True,
+            drop_last = True
+        ),
+    ),
+    
+    val_mode = dict(
+        transform = dict(
+            # RandomSizeAndCrop = {"size": 256, "crop_nopad": False},
+        ),
+        loader = dict(
+            batch_size = 8,
+            num_workers = 4,
+            pin_memory=True,
+            shuffle = False,
+            drop_last = False
+        )
+    ),
+
+    test_mode = dict(
+        transform = dict(
+            # RandomSizeAndCrop = {"size": 256, "crop_nopad": False},
+        ),
+        loader = dict(
+            batch_size = 8,
+            num_workers = 4,
+            pin_memory=True,
+            shuffle = False,
+            drop_last = False
+        )
+    ),
+)

configs/cdxformer.py

@@ -0,0 +1,77 @@
+net = 'CDXFormer'
+######################## base_config #########################
+epoch = 200
+gpus = [0]
+save_top_k = 1
+save_last = True
+check_val_every_n_epoch = 1
+logging_interval = 'epoch'
+resume_ckpt_path = None
+monitor_val = 'val_change_f1'
+monitor_test = ['test_change_f1']
+argmax = True
+
+test_ckpt_path = "work_dirs/CLCD_BS4_epoch200/CDXFormer/version_0/ckpts/test/test_change_f1/test_change_f1=0.9289-epoch=2.ckpt"
+
+exp_name = 'WHUCD_epoch200{}/{}'.format(epoch, net)
+
+######################## dataset_config ######################
+_base_ = [
+    './_base_/WHUCD_config.py',
+]
+num_class = 2
+ignore_index = 255
+
+######################### model_config #########################
+model_config = dict(
+        backbone = dict(
+            type = 'Base',
+            name = 'Seaformer',
+        ),
+        decoderhead = dict(
+            type = 'CDXLSTM',
+            channels = [64, 128, 192, 256]
+        )
+    )
+loss_config = dict(
+    type = 'myLoss',
+    loss_name = ['FocalLoss', 'dice_loss'],
+    loss_weight = [0.5, 0.5],
+    param = dict(
+        FocalLoss = dict(
+            gamma=0, 
+            alpha=None
+        ),
+        dice_loss = dict(
+            eps=1e-7
+        )
+    )
+)
+
+######################## optimizer_config ######################
+optimizer_config = dict(
+    optimizer = dict(
+        type = 'SGD',
+        lr = 0.05,
+        momentum=0.9,
+        weight_decay = 5e-5,
+        lr_mode = "single"
+    ),
+    scheduler = dict(
+        type = 'step',
+        step_size = 100,
+        gamma = 0.1
+    )
+)
+
+metric_cfg1 = dict(
+    task = 'multiclass',
+    average='micro',
+    num_classes = num_class, 
+)
+
+metric_cfg2 = dict(
+    task = 'multiclass',
+    average='none',
+    num_classes = num_class, 
+)