v/GFPGAN
1
0
Fork 0
mirror of https://github.com/TencentARC/GFPGAN.git synced 2025-05-19 00:30:16 -07:00
Code Issues Projects Releases Wiki Activity

update inference

Browse Source
This commit is contained in:
Xintao 2021-05-18 16:27:50 +08:00
parent d70570ef20
commit d9ec1351ff
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
inference_gfpgan_full.py
View File

@ -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,