我正在研究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是否足够?