From 74fcfea286a50ad5375077cf9a826a422a816860 Mon Sep 17 00:00:00 2001
From: Xintao <wxt1994@126.com>
Date: Wed, 28 Jul 2021 15:21:47 +0800
Subject: [PATCH] fix bug: extension

---
 inference_realesrgan.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/inference_realesrgan.py b/inference_realesrgan.py
index 46e84ee17..7de2414b0 100644
--- a/inference_realesrgan.py
+++ b/inference_realesrgan.py
@@ -28,7 +28,7 @@ def main():
         default='realesrgan',
         help='The upsampler for the alpha channels. Options: realesrgan | bicubic')
     parser.add_argument(
-        '--extension',
+        '--ext',
         type=str,
         default='auto',
         help='Image extension. Options: auto | jpg | png, auto means using the same extension as inputs')
@@ -101,10 +101,10 @@ def main():
             output_img[:, :, 3] = output_alpha
 
         # ------------------------------ save image ------------------------------ #
-        if args.extension == 'auto':
+        if args.ext == 'auto':
             extension = extension[1:]
         else:
-            extension == args.extension
+            extension = args.ext
         if img_mode == 'RGBA':  # RGBA images should be saved in png format
             extension = 'png'
         save_path = f'results/{imgname}_{args.suffix}.{extension}'