|
@@ -21,7 +21,7 @@ __all__ = [
|
|
|
'TestBITModel', 'TestCDNetModel', 'TestChangeStarModel', 'TestDSAMNetModel',
|
|
|
'TestDSIFNModel', 'TestFCEarlyFusionModel', 'TestFCSiamConcModel',
|
|
|
'TestFCSiamDiffModel', 'TestSNUNetModel', 'TestSTANetModel',
|
|
|
- 'TestChangeFormerModel'
|
|
|
+ 'TestChangeFormerModel', 'TestFCCDNModel'
|
|
|
]
|
|
|
|
|
|
|
|
@@ -225,3 +225,14 @@ class TestChangeFormerModel(TestCDModel):
|
|
|
dict(**base_spec, decoder_softmax=True),
|
|
|
dict(**base_spec, embed_dim=56)
|
|
|
] # yapf: disable
|
|
|
+
|
|
|
+
|
|
|
+class TestFCCDNModel(TestCDModel):
|
|
|
+ MODEL_CLASS = paddlers.rs_models.cd.FCCDN
|
|
|
+
|
|
|
+ def set_specs(self):
|
|
|
+ self.specs = [
|
|
|
+ dict(in_channels=3, num_classes=2),
|
|
|
+ dict(in_channels=8, num_classes=2),
|
|
|
+ dict(in_channels=3, num_classes=8)
|
|
|
+ ] # yapf: disable
|