如何从名称存储在pytorch中的csv文件中的文件夹中加载图像

Question

我有一个csv文件，其中包含必须从文件夹加载到数据加载器中的特定图像的名称。此csv文件还包含图像的标签/类。我尝试使用数据集类。我写了下面的代码：

class data(Dataset):
   def __init__(self,type='train',transform=None):
     self.imagecsv=''
     if type=='train':
       self.imagecsv=traincsv
     else:
       self.imagecsv=testcsv
   def __getitem__(self):
     imagelabels=self.imagecsv[['image_id','Labels']]
     images=[]
     for i in imagelabels['image_id']:
       image=imread(os.path.join(path,i+'.jpg'))
       images.append(image)
     imagelabels['images']=images
     imagelabels.drop(columns='image_id')
     return imagelabels  

当我尝试运行此代码时，系统每次都崩溃。我在带有GPU的Google colab上运行它。我知道我没有在 getitem 函数中添加索引参数，这是因为我需要在训练csv文件中指定了名称的所有图像。有没有更清洁的实现不会使我的系统崩溃的实现？我也尝试使用Image.open，但系统再次崩溃。