Skip to content
Snippets Groups Projects
Commit 4d7e2b88 authored by Bobholamovic's avatar Bobholamovic
Browse files

Add EF arch

parent 1d1980eb
Branches main
No related tags found
No related merge requests found
...@@ -104,6 +104,9 @@ def single_model_factory(model_name, C): ...@@ -104,6 +104,9 @@ def single_model_factory(model_name, C):
elif name == 'SIAMUNET_DIFF': elif name == 'SIAMUNET_DIFF':
from models.siamunet_diff import SiamUnet_diff from models.siamunet_diff import SiamUnet_diff
return SiamUnet_diff(C.num_feats_in, 2) return SiamUnet_diff(C.num_feats_in, 2)
elif name == 'EF':
from models.unet import Unet
return Unet(C.num_feats_in, 2)
else: else:
raise NotImplementedError("{} is not a supported architecture".format(model_name)) raise NotImplementedError("{} is not a supported architecture".format(model_name))
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment