Open yuwl798180 opened 5 years ago
可以和之前的手写图片一样,用训练好的模型预测然后输出结果即可。
你好,这个示例代码中,model返回的loss,而且维度是128112112,view后也对不上(3,224,224)。
请问能具体点告知下,如何把训练后的结果展示出来吗?
我用 output = new_model(input_img)
方式得到的维度是 128*112*112
你好,这个示例代码中,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()
你好,这个示例代码中,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
如题,问下训练完成后,从哪里输出迁移后的结果?
可以告知下示例代码吗?