估计加载图像数据集可能使用的RAM使用量

时间:2019-03-07 10:38:17

标签: python numpy opencv

我正在研究kaggle猫与狗的图像分类挑战。我的电脑目前只有8GB的RAM,因此使用其中的一些内存我大约有6GB的空间。

首先尝试加载包含12500张图像的图像test_set,它在硬盘上占用285 mbs。运行该脚本会耗尽所有可用的RAM,并且可能会使PC停顿下来,因此我在它超过95%的使用率之前将其停止。

import os

test_images = os.listdir('data/test')
print(len(test_images))
>>> 12500

test_images_data = []

for image in test_images:
    image_data = cv2.imread('data/test/' + image)

    #convert color from BGR to RGB
    image_data = cv2.cvtColor(image_data, cv2.COLOR_BGR2RGB)

    test_images.append(image_data)

是否有一种方法可以事先告知可能使用多少RAM,所以我知道例如增加8GB RAM是否足够?

0 个答案:

没有答案