diff --git a/src/impl/builders/model_builders.py b/src/impl/builders/model_builders.py index e972da29b573d146001745f630a94d87061efb07..0f141590baf9c54bcad02ec10d67d84583e6bcbf 100644 --- a/src/impl/builders/model_builders.py +++ b/src/impl/builders/model_builders.py @@ -36,4 +36,16 @@ def build_SiamUnet_conc_model(C): @MODELS.register_func('SiamUnet_conc_OSCD_model') def build_SiamUnet_conc_OSCD_model(C): from models.siamunet_conc import SiamUnet_conc - return SiamUnet_conc(13, 2) \ No newline at end of file + return SiamUnet_conc(13, 2) + + +@MODELS.register_func('FresUnet_model') +def build_FresUnet_model(C): + from models.fresunet import FresUNet + return FresUNet(6, 2) + + +@MODELS.register_func('FresUnet_OSCD_model') +def build_FresUnet_OSCD_model(C): + from models.fresunet import FresUNet + return FresUNet(26, 2) \ No newline at end of file diff --git a/src/models b/src/models index f3a8e382e3446ca2daca752c7e8f22f30291afff..4dd83231f25319a7ebb16cbfa9912541ceabac9a 160000 --- a/src/models +++ b/src/models @@ -1 +1 @@ -Subproject commit f3a8e382e3446ca2daca752c7e8f22f30291afff +Subproject commit 4dd83231f25319a7ebb16cbfa9912541ceabac9a