PyTorch笔记（二）

PyTorch加载数据

dataset 提供一种方式去获取数据及label

dataload为后面的网络提供不同数据形式

from torch.utils.data import Dataset

 控制台console可以显示属性，可以调试或测试

（把图片粘贴到项目文件下，（项目名上右键打开于explorer可以打开文件夹），右键复制路径）

img_path="C:\\Users\\ASUS\\PycharmProjects\\pythonProject2\\dataset\\train\\ants\\0013035.jpg"#粘贴到引号里，改双斜杠转义
img=Image.open(img_path) #读取图片
img.size #看属性
img.show()#看照片

dir_path="dataset/train/ants"
import os#和操作系统接口
img_path_list=os.listdir(dir_path)#生成一个列表
root_dir="dataset/train"
lable_dir="ants"
path=os.path.join(root_dir,lable_dir)#前两个地址连起来

完整代码

from torch.utils.data import Dataset
from PIL import Image
import os

class MyData(Dataset):#创建Mydata类，继承Dataset
 def __init__(self,root_dir,label_dir):#初始化类，为Class提供全局变量，为后面的函数提供量
    self.root_dir = root_dir#成为全局变量
    self.label_dir = label_dir
    self.path = os.path.join(self.root_dir,self.label_dir)#获得路径地址
    self.img_path=os.listdir(self.path)#获得全部列表

 def __getitem__(self,idx):
   img_name=self.img_path[idx]#读取一个图片,知道名称
   img_item_path=os.path.join(self.root_dir,self.label_dir,img_name)
   img=Image.open(img_item_path)
   label=self.label_dir
   return img,label

 def __len__(self):
  return len(self.img_path)

root_dir="dataset/train"#变量
ants_label_dir="ants"
bees_label_dir="bees"
ants_dataset = MyData(root_dir, ants_label_dir)
bees_dataset = MyData(root_dir, bees_label_dir)

train_dataset=ants_dataset + bees_dataset