init
Browse files
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)
|