CIFAR-10上的Triplet数据集加载与处理代码

import  torch
import os
import numpy as np
from PIL import Image
from torch.utils.data.dataset import Dataset
import os.path
import torch.utils.data
import torchvision.transforms as transforms


class DatasetProcessingCIFAR_10(Dataset):  # 初始化两个文件夹列表 Class 中的三个内置函数
    def __init__(self, data_path, img_filename, label_filename, transform=None):
        # self.img_path  img_filename label
        #DATA_DIR = 'D:\CIFAR-10'
        #DATABASE_FILE = 'database_img.txt'  # 2
        #DATABASE_LABEL = 'database_label.txt'  # 3
        self.img_path = data_path
        self.transform = transform
        # reading img file from file
        img_filepath = os.path.join(data_path, img_filename)  # os.path.join 文件之所以经常会出现问题的主要原因是python代码文件和调用文件不在一个文件夹下
        fp = open(img_filepath, 'r')  # 这里是open文件
        self.img_filename = [x.strip() for x in fp]  # 返回照片文件名 这个是list格式
        fp.close()
        label_filepath = os.path.join(data_path, label_filename)
        fp_label = open(label_filepath, 'r')
        labels = [int(x.strip()) for x in fp_label]
        fp_label.close()
        self.label = labels

    def __getitem__(self, index):