未验证 提交 67d7e415 编写于 作者: W wangna11BD 提交者: GitHub

fix mpr and doc (#550)

上级 1b4bcd61
...@@ -18,6 +18,8 @@ faceenhancer = FaceEnhancement() ...@@ -18,6 +18,8 @@ faceenhancer = FaceEnhancement()
img = faceenhancer.enhance_from_image(img) img = faceenhancer.enhance_from_image(img)
``` ```
Note: please convert the image to float type, currently does not support int8 type.
### Train (TODO) ### Train (TODO)
In the future, training scripts will be added to facilitate users to train more types of GPEN. In the future, training scripts will be added to facilitate users to train more types of GPEN.
......
...@@ -18,6 +18,8 @@ faceenhancer = FaceEnhancement() ...@@ -18,6 +18,8 @@ faceenhancer = FaceEnhancement()
img = faceenhancer.enhance_from_image(img) img = faceenhancer.enhance_from_image(img)
``` ```
注意:请将图片转为float类型输入,目前不支持int8类型
### 训练(TODO) ### 训练(TODO)
未来还将添加训练脚本方便用户训练出更多类型的 GPEN 人脸增强。 未来还将添加训练脚本方便用户训练出更多类型的 GPEN 人脸增强。
......
...@@ -110,7 +110,7 @@ class MPRPredictor(BasePredictor): ...@@ -110,7 +110,7 @@ class MPRPredictor(BasePredictor):
os.makedirs(self.output_path, exist_ok=True) os.makedirs(self.output_path, exist_ok=True)
task_path = os.path.join(self.output_path, self.task) task_path = os.path.join(self.output_path, self.task)
os.makedirs(task_path, exist_ok=True) 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): for image_file in tqdm(image_files):
img = self.read_image(image_file) img = self.read_image(image_file)
image_name = os.path.basename(image_file) image_name = os.path.basename(image_file)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册