pytorch dataset类和os库的使用

from torch.utils.data import Dataset
import cv2
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) ## 'hymenoptera_data/hymenoptera_data/train\\ants'
        self.img_path = os.listdir(self.path) #列出文件夹下所有的文件名
    
    def __getitem__(self, item):
        img_name = self.img_path[item]
        img_item_path = os.path.join(self.path,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/hymenoptera_data/train"
ants_label_dir="ants"
ants_dataset = MyDate(root_dir,ants_label_dir)

原来相对路径是当前文件相对于python文件夹的路径