search

JaimeTang / book-code

《深度学习之PyTorch实战计算机视觉》全书代码
128 stars 33 forks source link

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

Open yuwl798180 opened 5 years ago

yuwl798180 commented 5 years ago

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

可以告知下示例代码吗？

JaimeTang commented 5 years ago

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

yuwl798180 commented 5 years ago

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

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

yuwl798180 commented 5 years ago

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

Master-cai commented 5 years ago

你好，这个示例代码中，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 commented 4 years ago

你好，这个示例代码中，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