summaryrefslogtreecommitdiff
path: root/models/models.py
diff options
context:
space:
mode:
authorjunyanz <junyanzhu89@gmail.com>2018-03-27 19:41:45 -0400
committerjunyanz <junyanzhu89@gmail.com>2018-03-27 19:41:45 -0400
commitb77d9b593c9fa3b0db4e5e2a77c23777abac57ce (patch)
tree35faf40692b6379b101fdef1608eb49e7b2fff9b /models/models.py
parentbae3f780dce02baa928e684d92d674ac8acdd95d (diff)
code refactoring
Diffstat (limited to 'models/models.py')
-rw-r--r--models/models.py20
1 files changed, 0 insertions, 20 deletions
diff --git a/models/models.py b/models/models.py
deleted file mode 100644
index 39cc020..0000000
--- a/models/models.py
+++ /dev/null
@@ -1,20 +0,0 @@
-def create_model(opt):
- model = None
- print(opt.model)
- if opt.model == 'cycle_gan':
- assert(opt.dataset_mode == 'unaligned')
- from .cycle_gan_model import CycleGANModel
- model = CycleGANModel()
- elif opt.model == 'pix2pix':
- assert(opt.dataset_mode == 'aligned')
- from .pix2pix_model import Pix2PixModel
- model = Pix2PixModel()
- elif opt.model == 'test':
- assert(opt.dataset_mode == 'single')
- from .test_model import TestModel
- model = TestModel()
- else:
- raise ValueError("Model [%s] not recognized." % opt.model)
- model.initialize(opt)
- print("model [%s] was created" % (model.name()))
- return model