text_plain_output_1.png
import matplotlib.pyplot as plt

import matplotlib.pyplot as plt

plt.figure(figsize=(24, 18))
for i in range(0, 32):
    plt.subplot(8, 8, 2 * i + 1)
    plt.imshow(images_gray[i], cmap='gray')
    plt.title(f'Input {i + 1}')
    plt.axis('off')
    plt.subplot(8, 8, 2 * i + 2)
    plt.imshow(images_col[i])
    plt.title(f'Output {i + 1}')
    plt.axis('off')
plt.show()