Update run.py
Browse files
run.py
CHANGED
|
@@ -64,13 +64,13 @@ class Options():
|
|
| 64 |
def updateOptions(self, phase):
|
| 65 |
|
| 66 |
if phase == "correct_to_mask":
|
| 67 |
-
self.checkpoints_dir = "
|
| 68 |
|
| 69 |
elif phase == "maskref_to_maskdet":
|
| 70 |
-
self.checkpoints_dir = "
|
| 71 |
|
| 72 |
elif phase == "maskfin_to_nude":
|
| 73 |
-
self.checkpoints_dir = "
|
| 74 |
|
| 75 |
# process(cv_img, mode)
|
| 76 |
# return:
|
|
|
|
| 64 |
def updateOptions(self, phase):
|
| 65 |
|
| 66 |
if phase == "correct_to_mask":
|
| 67 |
+
self.checkpoints_dir = "cm.lib"
|
| 68 |
|
| 69 |
elif phase == "maskref_to_maskdet":
|
| 70 |
+
self.checkpoints_dir = "mm.lib"
|
| 71 |
|
| 72 |
elif phase == "maskfin_to_nude":
|
| 73 |
+
self.checkpoints_dir = "mn.lib"
|
| 74 |
|
| 75 |
# process(cv_img, mode)
|
| 76 |
# return:
|