2016-04-24 103 views
0

我试图扭曲输入到NN的图像,但我似乎无法弄清楚如何去做。没有太多的库似乎提供了我在找的东西。如何扭曲火炬中的图像

最好将其种类torch.FloatTensor

编辑工作:实际上,功能image.warp看起来很有希望,但warp_test.lua是不是特别有帮助。我只是希望通过不同数量

回答

1

这里歪斜x中的图像是一个简单的版本:

require 'torch' 
require 'image' 

local function skew(input, factor) 
    local w, h = input:size(3), input:size(2) 
    local y  = torch.range(0, h - 1):view(h, 1):expand(h, w) 
    local x  = torch.range(0, w - 1):view(1, w):expand(h, w) 
    local field = torch.Tensor(2, h, w) 
    field[1] = y 
    field[2] = torch.add(x, factor or 0, y) 
    return image.warp(input, field, "bilinear", false, "pad", 0) 
end 

local output = skew(image.lena(), 0.25)