Olá Cesar, desculpa pela demora no retorno.
No curso de Keras a instrutora utilizar o matplotlib para fazer essas plotagens, poderia verificar se te ajuda no problema que está enfrentando
Deep Learning parte 1: introdução com Keras
for imagem in range(10):
plt.subplot(2, 5, imagem+1)
plt.imshow(imagens_treino[imagem])
plt.title(nomes_de_classificacoes[identificacoes_treino[imagem]])
O ponto mais importante é utilizar a função imshow que vai utilizar as informações da imagem para fazer o plot. Pode encontrar mais detalhes na documentação.
Espero ter ajudado, mas qualquer duvida não hesite em perguntar.
Caso este post tenha lhe ajudado, por favor, marcar como solucionado ✓. Bons Estudos!