diff options
| author | junyanz <junyanzhu89@gmail.com> | 2018-03-27 19:41:45 -0400 |
|---|---|---|
| committer | junyanz <junyanzhu89@gmail.com> | 2018-03-27 19:41:45 -0400 |
| commit | b77d9b593c9fa3b0db4e5e2a77c23777abac57ce (patch) | |
| tree | 35faf40692b6379b101fdef1608eb49e7b2fff9b /models/models.py | |
| parent | bae3f780dce02baa928e684d92d674ac8acdd95d (diff) | |
code refactoring
Diffstat (limited to 'models/models.py')
| -rw-r--r-- | models/models.py | 20 |
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 |
