3D张量作为火炬的图像

Question

如何在火炬中将3D张量保存为图像？

我正在使用图片包：https://github.com/torch/image/blob/master/doc/saveload.md

将3D张量保存为图像：

  image.save('train100.jpg', trainData[100])

加载时：

x = image.load('train100.jpg'.jpg' )

我得到x与原始张量不同，保存的图像全是白色。

这是代码：

require 'torch'
require 'image'
local trainset = torch.load('cifar.torch/cifar10-train.t7')
local testset = torch.load('cifar.torch/cifar10-test.t7')

local trainData = trainset.data:float() 
local trainLabels = trainset.label:float():add(1)

testData = testset.data:float()
testLabels = testset.label:float():add(1)

image.save('train100.jpg', trainData[100])

x = image.load('train100.jpg' )
print(trainData[100])
print(x)

Answer 1

通过标准化张量来解决这个问题：

image.save('train100.jpg', trainData[100])

Answer 2

您可以使用torchvision.utils.save_image

执行此操作

检查here的功能参数。