如何按照数据帧中指定的顺序从一个文件夹中加载图像以训练卷积神经网络？

Question

The photos provided are in a single folder, and the dataframe from the csv file provides the relevant image path for each item as seen here

如果我分别加载每个文件，将没有足够的内存来加载所有文件，但是我似乎找不到找到使用Pytorch的DataLoader和ImageFolder加载它的方法。当照片分为各自的标签类别时，我以前曾将DataLoader和ImageFolder用于其他CNN任务。

我试图编写一个生成器函数以将图像输入到我的模型中，但是生成器似乎崩溃了。

def img_generator(paths, size, batch_size):
    img_array = []
    for path in paths:
        img = cv2.imread(path)
        img = cv2.resize(img, (size, size))
        img_array.append(img)
        if len(img_array) == batch_size: 
            yield np.array(img_array)
    yield np.array(img_array)

beauty_generator = img_generator(beauty_images, 128, 1)

for item in beauty_generator:
    print('ok')

这将输出'ok'一次，然后就挂在那里。

任何帮助将不胜感激！