pytorch dataset用来实现tensor转换

import numpy as np
from PIL import Image
import torch
from torchvision import transforms
from skimage.io import imread
a = np.ones([5,5,5])
b = a[1:1+1]#拆分
print(b.shape)

target1 = imread(r"D:\pku\new desktop\dataset-DIV2k\Set14\baboon.bmp")
target2 = Image.open(r"D:\pku\new desktop\dataset-DIV2k\Set14\baboon.bmp")
print('yes')

#在 dataset的getitem函数中，将对应的np.array转换为对应的tensor即可
trans = transforms.ToTensor()
target1_tensor = trans(target1)
target2_tensor = trans(target2)
#注意 这个地方 即使你读入的是uint8格式数据 他也会对应给你转换为float32，所以可以放心直接用
#如果不这样的话，数据需要先transpose，再除以255.0，最后转换为float32。

print(yes)

#上面即实现的源码 可以看出 对于一个numpy array图像 首先transpose，然后除以255.0并且float