機器學習圖像處理中的內存錯誤

Question

我正在構建一個圖像處理分類器，這是我代碼的一小部分，我的問題是第一行表明我有一個整數數組img_data_list 。 我將其轉換為float，但這給了我一個Memory Error 。

我有一個7 Gb和12 Gb RAM的數據集。

img_data = np.array(img_data_list)
img_data = img_data.astype('float32')
img_data /= 255
print (img_data.shape)

Answer 1

你在哪里困惑？ 您有一個7G數據集。 您的代碼需要三個副本：原始數據集，所述NumPy的陣列，並且那一個浮子轉換。 您沒有足夠的內存來創建第二個副本。

顯而易見的解決方案是在計算機上增加RAM。 您擁有的配置不足以解決您要解決的問題類別。 另一種可能性是在輸入時將數據轉換為float32 ，這樣RAM中將只有一個數據副本。