2016-11-20 1 views
0

Wie kann ich 3D Tensor als Bild in Fackel speichern?3D Tensor als Bild in Fackel

Ich verwende Bild Paket: https://github.com/torch/image/blob/master/doc/saveload.md

3D-Tensor als Bild zu speichern:

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

beim Laden zurück:

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

ich, dass x von dem unterscheidet Original Tensor und das gespeicherte Bild ist alles weiß.

Dies ist der Code:

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) 

Antwort

0

dies durch Normalisieren der Tensor gelöst wurde:

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