Update inference_gfpgan.py

inference_gfpgan.py

@@ -70,25 +70,25 @@ def main():
         # save faces
         for idx, (cropped_face, restored_face) in enumerate(zip(cropped_faces, restored_faces)):
             # save cropped face
-            save_crop_path = os.path.join(args.save_root, 'cropped_faces', f'{basename}_{idx:02d}.png')
+            save_crop_path = os.path.join(args.save_root, 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.save_root, 'restored_faces', save_face_name)
+            save_restore_path = os.path.join(args.save_root, save_face_name)
             imwrite(restored_face, save_restore_path)
             # save cmp image
             cmp_img = np.concatenate((cropped_face, restored_face), axis=1)
-            imwrite(cmp_img, os.path.join(args.save_root, 'cmp', f'{basename}_{idx:02d}.png'))
+            imwrite(cmp_img, os.path.join(args.save_root, f'{basename}_{idx:02d}.png'))
 
         # save restored img
         if restored_img is not None:
             if args.suffix is not None:
-                save_restore_path = os.path.join(args.save_root, 'restored_imgs', f'{basename}_{args.suffix}{ext}')
+                save_restore_path = os.path.join(args.save_root, f'{basename}_{args.suffix}{ext}')
             else:
-                save_restore_path = os.path.join(args.save_root, 'restored_imgs', img_name)
+                save_restore_path = os.path.join(args.save_root, img_name)
             imwrite(restored_img, save_restore_path)
 
     print(f'Results are in the [{args.save_root}] folder.')