Как я могу показать красочные изображения cifar10, используя matplotlib?

Question

Я пытаюсь показать сгенерированные изображения из набора данных cifar10 на рисунке (я сгенерировал изображения с использованием генеративных состязательных сетей. Но цвет изображений - зеленый!

cntr = 0
        for e in plot_epochs:
            if i == e:
                if row_counter < 10:
                    for q in range(10):
                        z = np.random.uniform(-1, 1, (100, cfg.z_shape))
                        imgs = mySess.run(generator, feed_dict={phZ: z})
                        imgs = imgs * 0.5 + 0.5
                        axs_result[q, row_counter].imshow(imgs[cntr, :, :, 0], cmap="viridis")
                        axs_result[q, row_counter].axis('off')
                        cntr += 1
                    row_counter += 1

            fig_result.savefig("results/worker_%d_results/%d.png" % (nID, nID))

Как я могу исправить проблему цвета изображения?