将图像和标签存储到hdf5文件

Question

我有一个图像文件夹和一个带有对应于每个图像标签的csv文件。我正在尝试训练pytorch模型，问题是数据加载似乎是一个瓶颈，因此我想将图像存储到HDf5文件中并读取它们。