TencentARC/GFPGAN
1
0
Fork 0
mirror of https://github.com/TencentARC/GFPGAN.git synced 2026-07-08 17:56:12 +00:00
Code Issues 476 Packages Projects Releases 8 Wiki Activity

add ext option

Browse Source
This commit is contained in:
Xintao
2021-09-10 18:46:45 +08:00
parent f6d3f70646
commit e5adc0dd06
1 changed files with 15 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
inference_gfpgan.py
View File

@@ -24,6 +24,12 @@ def main():
parser.add_argument('--aligned', action='store_true')
parser.add_argument('--paste_back', action='store_false')
parser.add_argument('--save_root', type=str, default='results')
parser.add_argument(
'--ext',
type=str,
default='auto',
help='Image extension. Options: auto | jpg | png, auto means using the same extension as inputs')
args = parser.parse_args()
args = parser.parse_args()
if args.test_path.endswith('/'):
@@ -85,10 +91,16 @@ def main():
# save restored img
if restored_img is not None:
if args.suffix is not None:
save_restore_path = os.path.join(args.save_root, 'restored_imgs', f'{basename}_{args.suffix}{ext}')
if args.ext == 'auto':
extension = ext[1:]
else:
save_restore_path = os.path.join(args.save_root, 'restored_imgs', img_name)
extension = args.ext
if args.suffix is not None:
save_restore_path = os.path.join(args.save_root, 'restored_imgs',
f'{basename}_{args.suffix}.{extension}')
else:
save_restore_path = os.path.join(args.save_root, 'restored_imgs', f'{basename}.{extension}')
imwrite(restored_img, save_restore_path)
print(f'Results are in the [{args.save_root}] folder.')