Dataset

提供一种方式去获取数据及其label（告诉我们数据在什么地方，有多少数据……）。

• 如何获取每一个数据及其label
• 告诉我们总共有多少的数据 ```python from torch.utils.data import Dataset from PIL import Image import os

class MyData(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, index):
    img_name = self.img_path[index]
    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 = “数据集/hymenoptera_data/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

```