Bladeren bron

Add unittest of FCCDN

Bobholamovic 2 jaren geleden
bovenliggende
commit
76e3fe35a6
1 gewijzigde bestanden met toevoegingen van 12 en 1 verwijderingen
  1. 12 1
      tests/rs_models/test_cd_models.py

+ 12 - 1
tests/rs_models/test_cd_models.py

@@ -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