Update RestoreFormer.py
Browse files- RestoreFormer.py +1 -1
RestoreFormer.py
CHANGED
@@ -2,7 +2,7 @@ import os
|
|
2 |
|
3 |
import cv2
|
4 |
import torch
|
5 |
-
from
|
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
|