add inputs folder

TencentARC · Jun 15, 2021 · 14e04ea · 14e04ea · HAMID4046 · Oct 13, 2022
1 parent 88fd720
commit 14e04ea
Show file tree

Hide file tree

Showing 7 changed files with 5 additions and 2 deletions.
diff --git 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 <results> folder.')
diff --git a/inputs/cropped_faces/Adele_crop.png b/inputs/cropped_faces/Adele_crop.png
diff --git a/inputs/cropped_faces/Julia_Roberts_crop.png b/inputs/cropped_faces/Julia_Roberts_crop.png
diff --git a/inputs/cropped_faces/Justin_Timberlake_crop.png b/inputs/cropped_faces/Justin_Timberlake_crop.png
diff --git a/inputs/cropped_faces/Paris_Hilton_crop.png b/inputs/cropped_faces/Paris_Hilton_crop.png
diff --git a/inputs/whole_imgs/00.jpg b/inputs/whole_imgs/00.jpg
diff --git a/inputs/whole_imgs/10045.png b/inputs/whole_imgs/10045.png