From 10ab8ea5b4e65f6f173c0966bd895529b8046b22 Mon Sep 17 00:00:00 2001 From: LielinJiang Date: Thu, 22 Oct 2020 07:51:23 +0000 Subject: [PATCH] fix apps output when input is image --- ppgan/apps/deoldify_predictor.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/ppgan/apps/deoldify_predictor.py b/ppgan/apps/deoldify_predictor.py index df54c9c..b3ca461 100644 --- a/ppgan/apps/deoldify_predictor.py +++ b/ppgan/apps/deoldify_predictor.py @@ -133,8 +133,10 @@ class DeOldifyPredictor(BasePredictor): else: pred_img = self.run_image(input) + out_path = None if self.output: - base_name = os.path.basename(input) - pred_img.save(os.path.join(self.output, base_name + '.png')) + base_name = os.path.splitext(os.path.basename(input))[0] + out_path = os.path.join(self.output, base_name + '.png') + pred_img.save(out_path) - return pred_img + return pred_img, out_path -- GitLab