完成了在普通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。我希望有人在网格关闭的情况下能帮助我。