Update inference_gfpgan.py

inference_gfpgan.py

@@ -134,20 +134,20 @@ def main():
             input_img, has_aligned=args.aligned, only_center_face=args.only_center_face, paste_back=True)
 
         # save faces
-        for idx, (cropped_face, restored_face) in enumerate(zip(cropped_faces, restored_faces)):
+     #   for idx, (cropped_face, restored_face) in enumerate(zip(cropped_faces, restored_faces)):
             # save cropped face
-           save_crop_path = os.path.join(args.output, 'cropped_faces', f'{basename}_{idx:02d}.png')
-           imwrite(cropped_face, save_crop_path)
+     #      save_crop_path = os.path.join(args.output, 'cropped_faces', f'{basename}_{idx:02d}.png')
+     #      imwrite(cropped_face, save_crop_path)
             # save restored face
-           if args.suffix is not None:
-                save_face_name = f'{basename}_{idx:02d}_{args.suffix}.png'
-           else:
-                save_face_name = f'{basename}_{idx:02d}.png'
-           save_restore_path = os.path.join(args.output, 'restored_faces', save_face_name)
-           imwrite(restored_face, save_restore_path)
+     #      if args.suffix is not None:
+     #           save_face_name = f'{basename}_{idx:02d}_{args.suffix}.png'
+     #      else:
+     #           save_face_name = f'{basename}_{idx:02d}.png'
+     #      save_restore_path = os.path.join(args.output, 'restored_faces', save_face_name)
+     #      imwrite(restored_face, save_restore_path)
             # save comparison image
-           cmp_img = np.concatenate((cropped_face, restored_face), axis=1)
-           imwrite(cmp_img, os.path.join(args.output, 'cmp', f'{basename}_{idx:02d}.png'))
+     #      cmp_img = np.concatenate((cropped_face, restored_face), axis=1)
+     #      imwrite(cmp_img, os.path.join(args.output, 'cmp', f'{basename}_{idx:02d}.png'))
           #  
         # save restored img
         if restored_img is not None: