Question

我试图将图像倾斜以输入NN，但我似乎无法弄清楚如何做到这一点。没有太多的图书馆可以提供我正在寻找的东西。

最好它适用于火炬类型.FloatTensor

编辑：实际上，函数image.warp看起来很有希望，但是warp_test.lua并不是特别有用。我只是想在x中扭曲图像

Answer 1

这是一个简单的版本：

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)

如何在Torch中扭曲图像

1 个答案: