pytorch
z80179030z
这个作者很懒,什么都没留下…
展开
-
Pytorch(二)
通过Dataset加载自己的数据集#加载自己的数据集dataset = Data.TensorDataset(data_tensor = x, target_tensor = y)loader = Data.Dataloader(dataset = dataset, batch_size = BATCH_SIZE, ....)# 这里的loader会返回两个值,feature_data 和 tar原创 2021-11-29 19:59:50 · 81 阅读 · 0 评论 -
Pytorch (一)
Transforms中几个常用的函数:ToTensor:把其他格式的数据转换为tensorfrom torchvision import transformswriter = SummaryWriter("logs")img = Image.open(path)tensor_trans = transforms.ToTensor()tensor_img = tensor_trans(img)Resize:改变图像尺寸。常根据已有神经网络的要求来改变输入图像的sizetrans_r原创 2021-11-23 22:44:00 · 1051 阅读 · 0 评论 -
Pytorch 图片的读取
图片的读取from PIL import Imageimg = Image.open(path)# 这样读取到的图片格式是PIL格式或者import cv2img = cv2.imread(path)# 这样读取的图片是np.array()格式自定义函数读取图片并储存到列表from torch.utils.data import Datasetfrom PIL import Imageimport osclass MyData(Dataset): def __i原创 2021-11-23 22:23:47 · 764 阅读 · 0 评论