jhj0517
commited on
Commit
•
ba273d9
1
Parent(s):
d10784a
Handle img type
Browse files
modules/image_restoration/real_esrgan/wrapper/real_esrganer.py
CHANGED
@@ -179,6 +179,9 @@ class RealESRGANer():
|
|
179 |
|
180 |
@torch.no_grad()
|
181 |
def enhance(self, img, outscale=None, alpha_upsampler='realesrgan'):
|
|
|
|
|
|
|
182 |
h_input, w_input = img.shape[0:2]
|
183 |
# img: numpy
|
184 |
img = img.astype(np.float32)
|
|
|
179 |
|
180 |
@torch.no_grad()
|
181 |
def enhance(self, img, outscale=None, alpha_upsampler='realesrgan'):
|
182 |
+
if isinstance(img, str):
|
183 |
+
img = cv2.imread(img)
|
184 |
+
|
185 |
h_input, w_input = img.shape[0:2]
|
186 |
# img: numpy
|
187 |
img = img.astype(np.float32)
|