hylee commited on
Commit
f306a44
1 Parent(s): d84f784
Files changed (1) hide show
  1. app.py +4 -18
app.py CHANGED
@@ -155,24 +155,10 @@ def run(
155
  get_68lm(imgfile, lmfile5, lmfile68, detector, predictor)
156
 
157
  imgs = []
158
- for part in ['eyel', 'eyer', 'nose', 'mouth']:
159
- savepath = os.path.join(opt.cmask_dir + part, name+'.png')
160
- get_partmask(imgfile, part, lmfile68, savepath)
161
- #imgs.append(savepath)
162
-
163
-
164
- # '''
165
- # 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
166
- # '''
167
- # opt.dataroot='dataset/test_single'
168
- # opt.name = 'apdrawinggan++_author'
169
- # opt.model = 'test'
170
- # opt.use_resnet = True
171
- # opt.netG = 'resnet_9blocks'
172
- # opt.which_epoch = 150
173
- # opt.how_many = 1000
174
- # opt.gpu_ids = '-1'
175
- # opt.imagefolder = 'images-single'
176
 
177
  data_loader = CreateDataLoader(opt)
178
  dataset = data_loader.load_data()
 
155
  get_68lm(imgfile, lmfile5, lmfile68, detector, predictor)
156
 
157
  imgs = []
158
+ # for part in ['eyel', 'eyer', 'nose', 'mouth']:
159
+ # savepath = os.path.join(opt.cmask_dir + part, name+'.png')
160
+ # get_partmask(imgfile, part, lmfile68, savepath)
161
+ # #imgs.append(savepath)
 
 
 
 
 
 
 
 
 
 
 
 
 
 
162
 
163
  data_loader = CreateDataLoader(opt)
164
  dataset = data_loader.load_data()