第八章图像风格迁移训练后的图片如何输出

[yuwl798180]

如题，问下训练完成后，从哪里输出迁移后的结果？

可以告知下示例代码吗？

[JaimeTang]

可以和之前的手写图片一样，用训练好的模型预测然后输出结果即可。

[yuwl798180]

你好，这个示例代码中，model返回的loss，而且维度是128112112，view后也对不上(3,224,224)。

请问能具体点告知下，如何把训练后的结果展示出来吗？

[yuwl798180]

我用 output = new_model(input_img) 方式得到的维度是 128*112*112

[Master-cai]

你好，这个示例代码中，model返回的loss，而且维度是128_112_112，view后也对不上(3,224,224)。

请问能具体点告知下，如何把训练后的结果展示出来吗？

你好，我自己写了一段代码把结果展示出来了，加在 while 结束以后就可以
下面是代码：
img = torchvision.utils.make_grid(input_img)
img = img.cpu()
img = img.numpy().transpose([1, 2, 0])
plt.imshow(img)
plt.show()

[leewi9]

你好，这个示例代码中，model 返回的 loss，而且维度是 128_112_112，view 后也对不上 (3,224,224)。
请问能具体点告知下，如何把训练后的结果展示出来吗？

你好，我自己写了一段代码把结果展示出来了，加在 while 结束以后就可以
下面是代码：
img = torchvision.utils.make_grid(input_img)
img = img.cpu()
img = img.numpy().transpose([1, 2, 0])
plt.imshow(img)
plt.show()

好的，可以加你吗，我的QQ2768881880