lucasrufkahr/Real-ESRGAN
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix bug: extension

Browse Source
This commit is contained in:
Xintao
2021-07-28 15:21:47 +08:00
parent da1e1ee805
commit 74fcfea286
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
inference_realesrgan.py
View File

@@ -28,7 +28,7 @@ def main():
default='realesrgan', default='realesrgan',
help='The upsampler for the alpha channels. Options: realesrgan | bicubic') help='The upsampler for the alpha channels. Options: realesrgan | bicubic')
parser.add_argument( parser.add_argument(
'--extension', '--ext',
type=str, type=str,
default='auto', default='auto',
help='Image extension. Options: auto | jpg | png, auto means using the same extension as inputs') 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 output_img[:, :, 3] = output_alpha
# ------------------------------ save image ------------------------------ # # ------------------------------ save image ------------------------------ #
if args.extension == 'auto': if args.ext == 'auto':
extension = extension[1:] extension = extension[1:]
else: else:
extension == args.extension extension = args.ext
if img_mode == 'RGBA': # RGBA images should be saved in png format if img_mode == 'RGBA': # RGBA images should be saved in png format
extension = 'png' extension = 'png'
save_path = f'results/{imgname}_{args.suffix}.{extension}' save_path = f'results/{imgname}_{args.suffix}.{extension}'