From b77d9b593c9fa3b0db4e5e2a77c23777abac57ce Mon Sep 17 00:00:00 2001 From: junyanz Date: Tue, 27 Mar 2018 19:41:45 -0400 Subject: code refactoring --- models/models.py | 20 -------------------- 1 file changed, 20 deletions(-) delete mode 100644 models/models.py (limited to 'models/models.py') 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 -- cgit v1.2.3-70-g09d2