Python 如何使用matplotlib显示彩色cifar10图像？

我试图在一个图中显示从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))

cntr=0
对于地块_时期的e：
如果i==e：
如果行计数器<10：
对于范围（10）内的q：
z=np.随机.均匀（-1,1，（100，cfg.z_形））
imgs=myess.run（生成器，feed_dict={phZ:z}）
imgs=imgs*0.5+0.5
axs_结果[q，行计数器].imshow（imgs[cntr，：，：，0]，cmap=“viridis”）
axs\U结果[q，行\U计数器]。轴（'off'）
cntr+=1
行计数器+=1
fig_result.savefig（“results/worker_%d_results/%d.png”%（nID，nID））

---

如何解决图像的颜色问题？

更改您使用的颜色贴图

cmap=“viridis”

，它确实是绿色的。如果您不喜欢，请使用其他颜色贴图；可用的颜色贴图位于