python引用相对路径

文件夹ants/bees文件夹与learn_data.py隶属于同一个目录data_process

 所以引用相对路径的方式即为：

class MyData(Dataset):

    def __init__(self, root_dir, label_dir):
        self.root_dir = root_dir            #根目录，即hymenoptera_data/train
        self.label_dir = label_dir          #标签目录，即ants或bees
        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)

ants_root_dir = "hymenoptera_data/train"　　#路径引用
ants_label_dir = "ants"
ants_dataset = MyData(ants_root_dir,ants_label_dir)

bees_root_dir = "hymenoptera_data/train"
bees_label_dir = "bees"
bees_dataset = MyData(bees_root_dir,bees_label_dir)