diff --git a/inference_gfpgan_full.py b/inference_gfpgan_full.py index 6b3d4119..504e5c2c 100644 --- a/inference_gfpgan_full.py +++ b/inference_gfpgan_full.py @@ -67,9 +67,10 @@ def restoration(gfpgan, face_helper, img_path, save_root, has_aligned=False, onl parser.add_argument('--upscale_factor', type=int, default=1) parser.add_argument('--model_path', type=str, default='experiments/pretrained_models/GFPGANv1.pth') - parser.add_argument('--test_path', type=str, default='inputs') + parser.add_argument('--test_path', type=str, default='inputs/whole_imgs') parser.add_argument('--suffix', type=str, default=None, help='Suffix of the restored faces') parser.add_argument('--only_center_face', action='store_true') + parser.add_argument('--aligned', action='store_true') args = parser.parse_args() if args.test_path.endswith('/'): @@ -107,6 +108,8 @@ def restoration(gfpgan, face_helper, img_path, save_root, has_aligned=False, onl face_helper, img_path, save_root, - has_aligned=False, + has_aligned=args.aligned, only_center_face=args.only_center_face, suffix=args.suffix) + + print('Results are in the folder.') diff --git a/inputs/cropped_faces/Adele_crop.png b/inputs/cropped_faces/Adele_crop.png new file mode 100644 index 00000000..afeb5557 Binary files /dev/null and b/inputs/cropped_faces/Adele_crop.png differ diff --git a/inputs/cropped_faces/Julia_Roberts_crop.png b/inputs/cropped_faces/Julia_Roberts_crop.png new file mode 100644 index 00000000..38c75c6e Binary files /dev/null and b/inputs/cropped_faces/Julia_Roberts_crop.png differ diff --git a/inputs/cropped_faces/Justin_Timberlake_crop.png b/inputs/cropped_faces/Justin_Timberlake_crop.png new file mode 100644 index 00000000..f4c118c5 Binary files /dev/null and b/inputs/cropped_faces/Justin_Timberlake_crop.png differ diff --git a/inputs/cropped_faces/Paris_Hilton_crop.png b/inputs/cropped_faces/Paris_Hilton_crop.png new file mode 100644 index 00000000..8d8ffc1a Binary files /dev/null and b/inputs/cropped_faces/Paris_Hilton_crop.png differ diff --git a/inputs/whole_imgs/00.jpg b/inputs/whole_imgs/00.jpg new file mode 100644 index 00000000..3df79f94 Binary files /dev/null and b/inputs/whole_imgs/00.jpg differ diff --git a/inputs/whole_imgs/10045.png b/inputs/whole_imgs/10045.png new file mode 100644 index 00000000..72032fb4 Binary files /dev/null and b/inputs/whole_imgs/10045.png differ