init

app.py

@@ -118,37 +118,52 @@ def run(
         detector,
         predictor,
 ) -> tuple[PIL.Image.Image,PIL.Image.Image,PIL.Image.Image,PIL.Image.Image,]:
-    dataroot = 'images/' + compress_UUID()
-    opt.dataroot = os.path.join(dataroot, 'src/')
-    os.makedirs(opt.dataroot, exist_ok=True)
-    opt.results_dir = os.path.join(dataroot, 'results/')
-    os.makedirs(opt.results_dir, exist_ok=True)
 
-    opt.lm_dir = os.path.join(dataroot, 'landmark/')
-    opt.cmask_dir = os.path.join(dataroot, 'mask/')
-    opt.bg_dir = os.path.join(dataroot, 'mask/bg')
-    os.makedirs(opt.lm_dir, exist_ok=True)
-    os.makedirs(opt.cmask_dir, exist_ok=True)
-    os.makedirs(opt.bg_dir, exist_ok=True)
+    # dataroot = 'images/' + compress_UUID()
+    # opt.dataroot = os.path.join(dataroot, 'src/')
+    # os.makedirs(opt.dataroot, exist_ok=True)
+    # opt.results_dir = os.path.join(dataroot, 'results/')
+    # os.makedirs(opt.results_dir, exist_ok=True)
+    #
+    # opt.lm_dir = os.path.join(dataroot, 'landmark/')
+    # opt.cmask_dir = os.path.join(dataroot, 'mask/')
+    # opt.bg_dir = os.path.join(dataroot, 'mask/bg')
+    # os.makedirs(opt.lm_dir, exist_ok=True)
+    # os.makedirs(opt.cmask_dir, exist_ok=True)
+    # os.makedirs(opt.bg_dir, exist_ok=True)
+    #
+    # shutil.copy(image.name, opt.dataroot)
+    #
+    # fullname = os.path.basename(image.name)
+    # name = fullname.split(".")[0]
+    #
+    # shutil.copyfile(mask.name, os.path.join(opt.bg_dir, name+'.png'))
+    #
+    # imgfile = os.path.join(opt.dataroot, fullname)
+    # lmfile5 = os.path.join(opt.lm_dir, name+'.txt')
+    # lmfile68 = os.path.join(opt.lm_dir, name + '_68.txt')
+    # # 预处理数据
+    # get_68lm(imgfile, lmfile5, lmfile68, detector, predictor)
+    #
+    # imgs = []
+    # for part in ['eyel', 'eyer', 'nose', 'mouth']:
+    #     savepath = os.path.join(opt.cmask_dir + part, name+'.png')
+    #     get_partmask(imgfile, part, lmfile68, savepath)
+    #     #imgs.append(savepath)
 
-    shutil.copy(image.name, opt.dataroot)
 
-    fullname = os.path.basename(image.name)
-    name = fullname.split(".")[0]
-
-    shutil.copyfile(mask.name, os.path.join(opt.bg_dir, name+'.png'))
-
-    imgfile = os.path.join(opt.dataroot, fullname)
-    lmfile5 = os.path.join(opt.lm_dir, name+'.txt')
-    lmfile68 = os.path.join(opt.lm_dir, name + '_68.txt')
-    # 预处理数据
-    get_68lm(imgfile, lmfile5, lmfile68, detector, predictor)
-
-    imgs = []
-    for part in ['eyel', 'eyer', 'nose', 'mouth']:
-        savepath = os.path.join(opt.cmask_dir + part, name+'.png')
-        get_partmask(imgfile, part, lmfile68, savepath)
-        #imgs.append(savepath)
+    '''
+    python test.py --dataroot dataset/test_single --name apdrawinggan++_author --model test --use_resnet --netG resnet_9blocks --which_epoch 150 --how_many 1000 --gpu_ids 0 --gpu_ids_p 0 --imagefolder images-single
+    '''
+    opt.dataroot='dataset/test_single'
+    opt.name = 'apdrawinggan++_author'
+    opt.model = 'test'
+    opt.use_resnet = True
+    opt.netG = 'resnet_9blocks'
+    opt.which_epoch = 150
+    opt.how_many = 1000
+    opt.gpu_ids = '-1'
+    opt.imagefolder = 'images-single'
 
     data_loader = CreateDataLoader(opt)
     dataset = data_loader.load_data()