2016-04-24 178 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)