本次内容主要是和大家分享，一起学习深度学习相关知识

需要使用到的相关库有Pytorch（最好是gpu，如果没有gpu那就使用cpu的）

Pillow库

内容一：数据加载

在这里的数据加载，就有两种方式可以选择：直接加载（Pillow），或者使用Pytorch加载

①直接加载

直接加载很简单，代码如下：

from PIL import Imageimg_path = 'data\1.jpg'
img = Image.open(img_path)
img.show()

②使用torch加载数据

代码如下：

import torch
from torch.utils.data import Dataset
form PIL import Image
import osclass Mydataset(Dataset):def __init__(self,root_path,label_path):self.root_path = root_pathself.label_patn = label_pathself.path = os.path.join(self.root_path,self.label.path) # 拼接，路径：root_path/label_pathself.img_path = os.listdir(self.path) # 该路径下所有图片def __getitem__(self,idx):self.img_name = self.img_path[idx]self.img_item_path = os.path.join(self.root_path,self.label,self.img_name)img = Image.open(img_item_path)label = label_pathreturn img,labeldef __len__(self):return len(self.img_name)root_path = ''
label_path = ''
dataset = Mydataset(root_path,label_path)
print(len(dataset))
img, label = dataset[0]
print(‘label’, label)
img.show