From 4d7e2b889c795101cceae7b1178ae70182249f81 Mon Sep 17 00:00:00 2001
From: Bobholamovic <bob1998425@hotmail.com>
Date: Fri, 13 Dec 2019 18:52:07 +0800
Subject: [PATCH] Add EF arch
---
src/core/factories.py | 3 +++
1 file changed, 3 insertions(+)
diff --git a/src/core/factories.py b/src/core/factories.py
index 652b776..2db268a 100644
--- a/src/core/factories.py
+++ b/src/core/factories.py
@@ -104,6 +104,9 @@ def single_model_factory(model_name, C):
elif name == 'SIAMUNET_DIFF':
from models.siamunet_diff import SiamUnet_diff
return SiamUnet_diff(C.num_feats_in, 2)
+ elif name == 'EF':
+ from models.unet import Unet
+ return Unet(C.num_feats_in, 2)
else:
raise NotImplementedError("{} is not a supported architecture".format(model_name))
--
GitLab