当前位置: 代码迷 >> 综合 >> pytorch dataset用来实现tensor转换
  详细解决方案

pytorch dataset用来实现tensor转换

热度:46   发布时间:2023-09-22 06:59:27.0
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)

pytorch dataset用来实现tensor转换

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

 

  相关解决方案