1 files changed, 15 insertions, 0 deletions
diff --git a/models/models.py b/models/models.py
new file mode 100644
index 0000000..7e790d0
--- /dev/null
+++ b/models/models.py
@@ -0,0 +1,15 @@
+
+def create_model(opt):
+    model = None
+    print(opt.model)
+    if opt.model == 'cycle_gan':
+        from .cycle_gan_model import CycleGANModel
+        assert(opt.align_data == False)
+        model = CycleGANModel()
+    if opt.model == 'pix2pix':
+        from .pix2pix_model import Pix2PixModel
+        assert(opt.align_data == True)
+        model = Pix2PixModel()
+    model.initialize(opt)
+    print("model [%s] was created" % (model.name()))
+    return model