xiexu911的博客

import torchimport torchvisionimport torchvision.transforms as transformimport torch.utils.data as Dataimport matplotlib.pyplot as pltfrom torch.utils.data import Dataset,DataLoaderimport pandas...

squeeze的用法主要就是对数据的维度进行压缩或者解压。先看torch.squeeze() 这个函数主要对数据的维度进行压缩，去掉维数为1的的维度，比如是一行或者一列这种，一个一行三列（1,3）的数去掉第一个维数为一的维度之后就变成（3）行。squeeze(a)就是将a中所有为1的维度删掉。不为1的维度没有影响。a.squeeze(N) 就是去掉a中指定的维数为一的维度。还有一种形式就是b=to...

import torchimport torchvisionimport torch.nn as nnimport torch.nn.functional as Fimport matplotlib.pyplot as pltimport numpy as np# data loading and transformingfrom torchvision.datasets impo...

这种读取图片的方式用的是torch自带的 ImageFolder，读取的文件夹必须在一个大的子文件下，按类别归好类。就像我现在要区分三个类别。#perpare data set#train datatrain_data=torchvision.datasets.ImageFolder('F:/eyeDataSet/trainData',transform=transforms.Com...