From 3fe47d14cd855f56edebf348cd3561a1b33da765 Mon Sep 17 00:00:00 2001
From: Bobholamovic <bob1998425@hotmail.com>
Date: Thu, 10 Dec 2020 20:00:09 +0800
Subject: [PATCH] Update submodule
---
src/impl/builders/model_builders.py | 14 +++++++++++++-
src/models | 2 +-
2 files changed, 14 insertions(+), 2 deletions(-)
diff --git a/src/impl/builders/model_builders.py b/src/impl/builders/model_builders.py
index e972da2..0f14159 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 f3a8e38..4dd8323 160000
--- a/src/models
+++ b/src/models
@@ -1 +1 @@
-Subproject commit f3a8e382e3446ca2daca752c7e8f22f30291afff
+Subproject commit 4dd83231f25319a7ebb16cbfa9912541ceabac9a
--
GitLab