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)
|