imshow()不想在tf会话中隐藏whitegrid

时间:2019-03-27 17:35:05

标签: python tensorflow matplotlib

完成了在普通Tensorflow中训练CNN的操作,但是如果没有白色网格,我将无法给出可视化效果

# Show Sample Predictions
predictions = sess.run(tf.argmax(CNN(testX[:25], weights, biases), axis=1))
f, axarr = plt.subplots(5, 5, figsize=(20,20))
for idx in range(25):
    axarr[int(idx/5), idx%5].imshow(np.squeeze(testX[idx]), cmap='gray')
    axarr[int(idx/5), idx%5].set_title(str(predictions[idx]), fontsize=30)

这只是mnist数据集的简单CNN。我希望有人在网格关闭的情况下能帮助我。

0 个答案:

没有答案