We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents 9c1f18b + 88a6801 commit 2f4558aCopy full SHA for 2f4558a
1 file changed
src/super_image/data/loader.py
@@ -11,7 +11,7 @@ class ImageLoader:
11
def load_image(image: Image):
12
lr = np.array(image.convert('RGB'))
13
lr = lr[::].astype(np.float32).transpose([2, 0, 1]) / 255.0
14
- return torch.as_tensor(np.array([lr]))
+ return torch.as_tensor(np.array(np.array([lr])))
15
16
@staticmethod
17
def _process_image_to_save(pred: Tensor):
0 commit comments