Update mpr_predictor.py

fixed MPR model predictor
3 years ago · f0109ae8a8
parent 099d5ac8c5
commit f0109ae8a8
1 changed files with 1 additions and 1 deletions
--- a/paddlers/models/ppgan/apps/mpr_predictor.py
+++ b/paddlers/models/ppgan/apps/mpr_predictor.py
@ -116,7 +116,7 @@ class MPRPredictor(BasePredictor):
        for image_file in tqdm(image_files):
            img = self.read_image(image_file)
            image_name = os.path.basename(image_file)
-            img.save(os.path.join(task_path, image_name))
+            # img.save(os.path.join(task_path, image_name))
            tmps = image_name.split('.')
            assert len(
                tmps) == 2, f'Invalid image name: {image_name}, too much "."'