未验证 提交 af0cd756 编写于 作者: L lijianshe02 提交者: GitHub

fix errors in face pase (#182)

上级 ac9e026d
......@@ -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
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册