matplotlibを使用してCSVからイメージを保存するMNIST

Question

MNISTデータセットでは、イメージはCSV形式であり、784列のそれぞれはピクセル強度に対応しています。私はimshowでそれらを見ることなく、これらの画像のそれぞれを保存したいです。

import numpy as np 
import csv 
import matplotlib.pyplot as plt 
i=0 
with open('Book1.csv', 'r') as csv_file: 
for data in csv.reader(csv_file): 

    # The rest of columns are pixels 
    pixels = data[:] 

    # This array will be of 1D with length 784 
    # The pixel intensity values are integers from 0 to 255 
    pixels = np.array(pixels, dtype='uint8') 

    # Reshape the array into 28 x 28 array (2-dimensional array) 
    pixels = pixels.reshape((28, 28)) 
    i +=1 
    # Plot 
    plt.title('Label is {label}'.format(label=label)) 
    plt.imshow(pixels, cmap='gray') 
    plt.savefig(str(i))' 

これで私は各画像を保存できません。

Answer 1

したい拡張子で終わる必要があるファイルの名前、

plt.savefig(str(i)+".png") 

またはあなたがformat指定する必要があります：あなたがに実行することが

plt.savefig(str(i), format="png") 

1つの追加の問題は、後にすべての画像が重ねて配置されるため、プロットが遅くなります。これを回避するには、ループの最後にplt.close()と呼ぶことができます。