dateset

dateset的含义

dateset是提供一种方式去加载数据，作用是获取每一个数据和对应的label，对数据进行统计，输出长度

dateset的使用

加载一张图片

from PIL import Image

import os

class MyDate(Dataset):

    def __init__(self, root_dir, label_dir):
        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)#打开图片，PIL格式
        label = self.label_dir
        return img, label
    def __len__(self):
        return len(self.img_path)


root_dir = "shuju/train"
test_label_dit = "train2017"
test_dataset = MyDate(root_dir, test_label_dit)
img,label = test_dataset[0]
img.show()

这样可以加载指定目录下的图片