init

app.py

@@ -8,6 +8,7 @@ import pathlib
 import sys
 from typing import Callable
 import uuid
+import cv2
 
 sys.path.insert(0, 'APDrawingGAN2')
 
@@ -125,51 +126,53 @@ def run(
         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)
-    #
-    # 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)
+    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)
 
-    '''
-    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'
+    shutil.copy(image.name, opt.dataroot)
+
+    fullname = os.path.basename(image.name)
+    name = fullname.split(".")[0]
+
+    bg = cv2.cvtColor(cv2.imread(mask.name), cv2.COLOR_BGR2GRAY)
+    cv2.imwrite(os.path.join(opt.bg_dir, name+'.png'), bg)
+    #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()