update

2023-06-01 23:32:08 +08:00
parent 7a6c92ae69
commit 70b22c4e4b
8 changed files with 77 additions and 15 deletions
@@ -7,13 +7,9 @@ from .config import device, num_classes


 def create_model(opt):
-    if opt.model == 'pix2pixHD':
-        #from .pix2pixHD_model import Pix2PixHDModel, InferenceModel
-        from .fs_model import fsModel
-        model = fsModel()
-    else:
-        from .ui_model import UIModel
-        model = UIModel()
+    #from .pix2pixHD_model import Pix2PixHDModel, InferenceModel
+    from .fs_model import fsModel
+    model = fsModel()

    model.initialize(opt)
    if opt.verbose: