卷积神经网络|导入图片

在学习卷积神经网络时,我们通常使用的就是公开的数据集,这里,我们不使用公开数据集,直接导入自己的图片数据,下面,就简单写个程序实现批量图片的导入。

import osfrom PIL import Imageimport numpy as npimport torchpath='E:\\3-10\\cat'IMG=[]filenames=[name for name in os.listdir(path)]for i,filename in enumerate(filenames):    img=Image.open(os.path.join(path,filename))    img=img.resize((28,28))#将图片像素改为28x28    img=np.array(img)#将图像数据转为numpy    img=torch.from_numpy(img)#将numpy转换为tensor张量    img=img.permute(2,0,1)#将H,W,C转换为C,H,W    IMG.append(img)#得到图片列表IMGEND=torch.stack([ig for ig in IMG],dim=0)#堆叠tensor​​​​​​​​​​​​​​
>>> IMGEND.size()torch.Size([5, 3, 28, 28])

借助上述代码,我们便可将我们准备的图片带入计算机,以便开展接下来的任务。

注:对上述代码稍加修改,便可引入标签,修改图片大小,分为训练集和测试集等等操作。

卷积神经网络|导入图片_第1张图片

你可能感兴趣的:(pytorch,cnn,人工智能,神经网络)