From af0cd756b9ffbedf2667410d87e1aa52045cc2de Mon Sep 17 00:00:00 2001
From: lijianshe02 <48898730+lijianshe02@users.noreply.github.com>
Date: Fri, 26 Feb 2021 14:13:11 +0800
Subject: [PATCH] fix errors in face pase (#182)

---
 ppgan/apps/face_parse_predictor.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/ppgan/apps/face_parse_predictor.py b/ppgan/apps/face_parse_predictor.py
index 4cf0084..f989060 100644
--- a/ppgan/apps/face_parse_predictor.py
+++ b/ppgan/apps/face_parse_predictor.py
@@ -52,8 +52,8 @@ class FaceParsePredictor(BasePredictor):
         mask = cv2.resize(mask.numpy(), (256, 256))
         mask = mask.astype(np.uint8)
         mask = mask2image(mask)
-        if not os.path.exists(output_path):
-            os.makedirs(output_path)
+        if not os.path.exists(self.output_path):
+            os.makedirs(self.output_path)
         save_path = os.path.join(self.output_path, 'face_parse.png')
         cv2.imwrite(save_path, mask)
         return mask
-- 
GitLab