hylee commited on
Commit
842a8b7
·
1 Parent(s): 1c1d3dc
Files changed (1) hide show
  1. app.py +5 -0
app.py CHANGED
@@ -120,6 +120,11 @@ def run(
120
  opt.results_dir = os.path.join(dataroot, 'results/')
121
  os.makedirs(opt.results_dir, exist_ok=True)
122
 
 
 
 
 
 
123
  shutil.copy(image.name, opt.dataroot)
124
 
125
  data_loader = CreateDataLoader(opt)
 
120
  opt.results_dir = os.path.join(dataroot, 'results/')
121
  os.makedirs(opt.results_dir, exist_ok=True)
122
 
123
+ opt.lm_dir = os.path.join(dataroot, 'landmark/')
124
+ opt.bg_dir = os.path.join(dataroot, 'mask/')
125
+ os.makedirs(opt.lm_dir, exist_ok=True)
126
+ os.makedirs(opt.bg_dir, exist_ok=True)
127
+
128
  shutil.copy(image.name, opt.dataroot)
129
 
130
  data_loader = CreateDataLoader(opt)