diff --git a/docs/en_US/tutorials/face_enhancement.md b/docs/en_US/tutorials/face_enhancement.md index 2c3b87b130b88378fc07364c20bd30aa1f57b409..318c010483fdd7b29ffaa7f6c88f5000f9994672 100644 --- a/docs/en_US/tutorials/face_enhancement.md +++ b/docs/en_US/tutorials/face_enhancement.md @@ -18,6 +18,8 @@ faceenhancer = FaceEnhancement() img = faceenhancer.enhance_from_image(img) ``` +Note: please convert the image to float type, currently does not support int8 type. + ### Train (TODO) In the future, training scripts will be added to facilitate users to train more types of GPEN. diff --git a/docs/zh_CN/tutorials/face_enhancement.md b/docs/zh_CN/tutorials/face_enhancement.md index 3bb9aceb6bf4c347b22d5a5c03841faab0d6beb5..0d3c3e166879f66be8cd6ec9972d2f053c28ccb5 100644 --- a/docs/zh_CN/tutorials/face_enhancement.md +++ b/docs/zh_CN/tutorials/face_enhancement.md @@ -18,6 +18,8 @@ faceenhancer = FaceEnhancement() img = faceenhancer.enhance_from_image(img) ``` +注意:请将图片转为float类型输入,目前不支持int8类型 + ### 训练(TODO) 未来还将添加训练脚本方便用户训练出更多类型的 GPEN 人脸增强。 diff --git a/ppgan/apps/mpr_predictor.py b/ppgan/apps/mpr_predictor.py index f9560e0c01b115d3f1a194b8bfd0bec996b15196..6ae12df54921f8a7186cfcb76edccef1f799f8bf 100644 --- a/ppgan/apps/mpr_predictor.py +++ b/ppgan/apps/mpr_predictor.py @@ -110,7 +110,7 @@ class MPRPredictor(BasePredictor): os.makedirs(self.output_path, exist_ok=True) task_path = os.path.join(self.output_path, self.task) os.makedirs(task_path, exist_ok=True) - image_files = self.get_images(self.images_path) + image_files = self.get_images(images_path) for image_file in tqdm(image_files): img = self.read_image(image_file) image_name = os.path.basename(image_file)