From d9ec1351ff43f4eb4c0f86fc9c1d1eac3f8aded5 Mon Sep 17 00:00:00 2001 From: Xintao Date: Tue, 18 May 2021 16:27:50 +0800 Subject: [PATCH] update inference --- inference_gfpgan_full.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/inference_gfpgan_full.py b/inference_gfpgan_full.py index 06b1d3f..6b3d411 100644 --- a/inference_gfpgan_full.py +++ b/inference_gfpgan_full.py @@ -100,8 +100,7 @@ if __name__ == '__main__': face_helper = FaceRestoreHelper( upscale_factor=1, face_size=512, crop_ratio=(1, 1), det_model='retinaface_resnet50', save_ext='png') - # scan all the jpg and png images - img_list = sorted(glob.glob(os.path.join(args.test_path, '*.[jp][pn]g'))) + img_list = sorted(glob.glob(os.path.join(args.test_path, '*'))) for img_path in img_list: restoration( gfpgan,