Incorrect prediction or visualisation

[CyberDreamer]

Hello.

After run this code:

output = autoencoder.predict_proba(train_data[8:9])
out_labels = np.argmax(output[0], axis=1)

pred = visualize(out_labels.reshape((height,width)), False)
plt.imshow(pred)
plt.figure(2)
plt.imshow(gt[8])
plt.show()

i have next result:

I change only Batch size and data size params. I work in PyCharm. It is incorrect prediction or visualisation? What i do wrong?