fix parse
Browse files
facelib/utils/face_restoration_helper.py
CHANGED
|
@@ -133,8 +133,9 @@ class FaceRestoreHelper(object):
|
|
| 133 |
|
| 134 |
# init face parsing model
|
| 135 |
self.use_parse = use_parse
|
| 136 |
-
self.
|
| 137 |
-
|
|
|
|
| 138 |
|
| 139 |
def set_upscale_factor(self, upscale_factor):
|
| 140 |
self.upscale_factor = upscale_factor
|
|
|
|
| 133 |
|
| 134 |
# init face parsing model
|
| 135 |
self.use_parse = use_parse
|
| 136 |
+
if self.use_parse:
|
| 137 |
+
self.face_parse = init_parsing_model(
|
| 138 |
+
model_name='parsenet', device=self.device)
|
| 139 |
|
| 140 |
def set_upscale_factor(self, upscale_factor):
|
| 141 |
self.upscale_factor = upscale_factor
|