wzhouxiff commited on
Commit
29adef7
1 Parent(s): 92e76e8

Update RestoreFormer.py

Browse files
Files changed (1) hide show
  1. RestoreFormer.py +1 -1
RestoreFormer.py CHANGED
@@ -2,7 +2,7 @@ import os
2
 
3
  import cv2
4
  import torch
5
- from basicsr.utils import img2tensor, tensor2img
6
  from basicsr.utils.download_util import load_file_from_url
7
  from facexlib.utils.face_restoration_helper import FaceRestoreHelper
8
  from torchvision.transforms.functional import normalize
 
2
 
3
  import cv2
4
  import torch
5
+ from img_utils import img2tensor, tensor2img
6
  from basicsr.utils.download_util import load_file_from_url
7
  from facexlib.utils.face_restoration_helper import FaceRestoreHelper
8
  from torchvision.transforms.functional import normalize