mirror of
https://github.com/TencentARC/GFPGAN.git
synced 2025-05-19 08:40:34 -07:00
update inference
This commit is contained in:
parent
d70570ef20
commit
d9ec1351ff
@ -100,8 +100,7 @@ if __name__ == '__main__':
|
|||||||
face_helper = FaceRestoreHelper(
|
face_helper = FaceRestoreHelper(
|
||||||
upscale_factor=1, face_size=512, crop_ratio=(1, 1), det_model='retinaface_resnet50', save_ext='png')
|
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, '*')))
|
||||||
img_list = sorted(glob.glob(os.path.join(args.test_path, '*.[jp][pn]g')))
|
|
||||||
for img_path in img_list:
|
for img_path in img_list:
|
||||||
restoration(
|
restoration(
|
||||||
gfpgan,
|
gfpgan,
|
||||||
|
Loading…
x
Reference in New Issue
Block a user