python - 无法在 google colab 上将图像转换为数组

Question

我需要帮助。我正在使用 google colab 来训练一个 CNN 模型。我的 train_set 中有 9101 张图像，但是当我尝试将图像转换为训练集的数组时出现问题。

这是我的代码：

def convert_image_to_array(files):
    array_img = []
    for file in files:
        array_img.append(img_to_array(load_img(file)))
    return np.array(array_img)
X_train = convert_image_to_array(X_train)
print(X_train.shape)

它总是无法转换图像，并且由于在使用所有可用 RAM 后崩溃而导致 google collab 不断重启。相同的代码适用于大小仅为 2277 个图像的 test_set。

有人可以帮我解决这个问题吗？所以它在谷歌合作上运行良好？

谢谢

标签： pythonnumpy