diff --git a/ppgan/apps/deoldify_predictor.py b/ppgan/apps/deoldify_predictor.py
index df54c9c797e90de305fb1a6b74371cddf466c140..b3ca4613bc38f216f321c79c1f768a82668edd4f 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