在python中读取图像时出现MemoryError

时间:2020-07-16 06:13:52

标签: python python-imaging-library

我想用python训练CNN模型,所以我要加载100000张图像(320x240),读取其RGB值并将其转换为numpy矩阵。但是,我每次都用光内存。有人可以告诉我如何解决这个问题吗?我认为最直接的方法就是释放我的记忆。

segments_lst = []
lst = sorted(glob.glob('tv7789/segments/*/*/*.jpg'))
for i in range(0,len(lst)-1):
    x = imageio.imread(lst[i]) 
    data = np.asarray(x)
    data = np.transpose(x, (2, 1, 0))
    segments_lst.append(data)
    gc.collect() #This does not help much.

0 个答案:

没有答案