3D 张量作为火炬中的图像答案

【问题标题】：3D tensor as an image in torch3D 张量作为火炬中的图像
【发布时间】：2017-04-04 04:08:25
【问题描述】：

如何将 3D 张量保存为 Torch 中的图像？

我正在使用图片包：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)

【问题讨论】：

标签： lua torch

【解决方案1】：

这是通过对张量进行归一化来解决的：

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

【讨论】：

【解决方案2】：

你可以使用torchvision.utils.save_image来做到这一点

检查here的函数参数。

【讨论】：