diff --git a/ppgan/apps/face_parse_predictor.py b/ppgan/apps/face_parse_predictor.py index 4cf0084e1821f45497ae8712f2804b39702bd238..f9890605c9229831c3fa00e0dae3a2abe3c96a19 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