Skip to content

Commit 2f4558a

Browse files
authored
Merge pull request #13 from tanndlin/main
Use np.array on image_load
2 parents 9c1f18b + 88a6801 commit 2f4558a

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

src/super_image/data/loader.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@ class ImageLoader:
1111
def load_image(image: Image):
1212
lr = np.array(image.convert('RGB'))
1313
lr = lr[::].astype(np.float32).transpose([2, 0, 1]) / 255.0
14-
return torch.as_tensor(np.array([lr]))
14+
return torch.as_tensor(np.array(np.array([lr])))
1515

1616
@staticmethod
1717
def _process_image_to_save(pred: Tensor):

0 commit comments

Comments
 (0)