Add unittest of FCCDN

own
Bobholamovic 2 years ago
parent b3e0c2eaf0
commit 76e3fe35a6
  1. 13
      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
Loading…
Cancel
Save