From 76e3fe35a665e69b0707213ff0b4d4184d502f1c Mon Sep 17 00:00:00 2001 From: Bobholamovic Date: Tue, 30 Aug 2022 20:48:48 +0800 Subject: [PATCH] Add unittest of FCCDN --- tests/rs_models/test_cd_models.py | 13 ++++++++++++- 1 file changed, 12 insertions(+), 1 deletion(-) diff --git a/tests/rs_models/test_cd_models.py b/tests/rs_models/test_cd_models.py index 3b3e683..e274100 100644 --- a/tests/rs_models/test_cd_models.py +++ b/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 \ No newline at end of file