将图像输入到网络进行训练后,得到的数据类型为output=(tensor([1,3,512,512])
将风格转化后的图像保存到本地后提示 应为tensor类型 [保存的函数为torch.utils.save_image()]
修改: 应将数据类型变为tensor([1,3,512,512]) , 使用 output[0] 输出即可
将图像输入到网络进行训练后,得到的数据类型为output=(tensor([1,3,512,512])
将风格转化后的图像保存到本地后提示 应为tensor类型 [保存的函数为torch.utils.save_image()]
修改: 应将数据类型变为tensor([1,3,512,512]) , 使用 output[0] 输出即可