我正在可视化原始图像和重建的图像。
我正在研究Chainer Vae模型,经过训练后,我试图通过功能save_image()可视化原始图像和重建的图像。
TypeError Traceback (most recent call last)
<ipython-input-42-57b6723a7a9f> in <module>()
2 x1 = model(x)
3 save_images(x.data, 'train')
----> 4 save_images(x1.data, 'train_reconstructed')
6 test_ind = [3, 2, 1, 18, 4, 8, 11, 17, 61]
<ipython-input-39-bdc1ae5cd577> in save_images(x, filename)
6 fig, ax = plt.subplots(3, 3, figsize=(9, 9), dpi=100)
7
----> 8 for ai, xi in zip(ax.flatten(), x):
9 ai.imshow(xi.reshape(512,512,4))
10 fig.savefig(filename)
TypeError:zip参数2必须支持迭代
原始图像可视化,但重建图像不可见。