如何批量读取文件夹中的文件:
1,先读一个文件,对比其它文件看有什么规律
zero = plt.imread('../data/data/0/0_1.bmp')
plt.imshow(zero,cmap='gray')
2,发现文件名相对应的数字编号规律,读取所有文件
data = []
target = []
for i in range(10):
for j in range(1,501):
temp = plt.imread(f'../data/data/{i}/{i}_{j}.bmp')
data.append(temp)
target.append(i)
中间过程省略**…**
部分真实与预测的结果对比画图展示
plt.figure(figsize=(10*2,10*2.5))
for i in range(100):
axes = plt.subplot(10,10,i+1)
axes.imshow(X_test[i*10].reshape(28,28),cmap='gray')
if y_[i*10] != y_test[i*10]:
axes.set_title('True:%s\nPredict:%s'%(y_test[i*10],y_[i*10]),fontdict=dict(fontsize=15,color='r'))
else:
axes.set_title('True:%s\nPredict:%s'%(y_test[i*10],y_[i*10]))
axes.axis('off')