support fc-ef fc-siam-conc fc-siam-diff

Author: likyoo

README.md

@@ -22,10 +22,12 @@ Toolbox:
 
 Change detection model:
 
-- [x] [Changer (arXiv'2022)](configs/changer)
-- [x] [FC-Siam-Diff (ICIP'2018)](configs/fcsn)
+- [x] [FC-EF (ICIP'2018)](configs/fcsn)
+- [x] [FC-Siam-diff (ICIP'2018)](configs/fcsn)
+- [x] [FC-Siam-conc (ICIP'2018)](configs/fcsn)
 - [x] [SNUNet (GRSL'2021)](configs/snunet)
 - [x] [BiT (TGRS'2021)](configs/bit)
+- [x] [Changer (arXiv'2022)](configs/changer)
 - [ ] ...
 
 The code of some models are borrowed directly from their official repositories.

configs/_base_/models/fc_ef.py

@@ -0,0 +1,23 @@
+# model settings
+norm_cfg = dict(type='SyncBN', requires_grad=True)
+base_channels = 16
+model = dict(
+    type='DIEncoderDecoder',
+    pretrained=None,
+    backbone=dict(
+        type='FC_EF',
+        in_channels=6,
+        base_channel=base_channels),
+    decode_head=dict(
+        type='FCNHead',
+        in_channels=base_channels,
+        channels=base_channels,
+        in_index=-1,
+        num_convs=0,
+        concat_input=False,
+        num_classes=2,
+        loss_decode=dict(
+            type='CrossEntropyLoss', use_sigmoid=False, loss_weight=1.0)),
+    # model training and testing settings
+    train_cfg=dict(),
+    test_cfg=dict(mode='whole'))

configs/_base_/models/fc_siam_conc.py

@@ -0,0 +1,23 @@
+# model settings
+norm_cfg = dict(type='SyncBN', requires_grad=True)
+base_channels = 16
+model = dict(
+    type='DIEncoderDecoder',
+    pretrained=None,
+    backbone=dict(
+        type='FC_Siam_conc',
+        in_channels=3,
+        base_channel=base_channels),
+    decode_head=dict(
+        type='FCNHead',
+        in_channels=base_channels,
+        channels=base_channels,
+        in_index=-1,
+        num_convs=0,
+        concat_input=False,
+        num_classes=2,
+        loss_decode=dict(
+            type='CrossEntropyLoss', use_sigmoid=False, loss_weight=1.0)),
+    # model training and testing settings
+    train_cfg=dict(),
+    test_cfg=dict(mode='whole'))

configs/_base_/models/fc_siam_diff.py

@@ -5,7 +5,7 @@
     type='DIEncoderDecoder',
     pretrained=None,
     backbone=dict(
-        type='SiamUnet_diff',
+        type='FC_Siam_diff',
         in_channels=3,
         base_channel=base_channels),
     decode_head=dict(

opencd/models/backbones/__init__.py

@@ -1,6 +1,7 @@
 from .interaction_resnet import IA_ResNetV1c
 from .interaction_resnest import IA_ResNeSt
-from .siamunet_diff import SiamUnet_diff
+from .fcsn import FC_EF, FC_Siam_diff, FC_Siam_conc
 from .snunet import SNUNet_ECAM
 
-__all__ = ['IA_ResNetV1c', 'IA_ResNeSt', 'SiamUnet_diff', 'SNUNet_ECAM']
+__all__ = ['IA_ResNetV1c', 'IA_ResNeSt', 'FC_EF', 'FC_Siam_diff', 
+           'FC_Siam_conc', 'SNUNet_ECAM']