Update inference_gfpgan.py

inference_gfpgan.py

@@ -134,21 +134,21 @@ 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:
             if args.ext == 'auto':