使用torch的image.display时保留纵横比

Question

我有以下用lua编写的非常简单的脚本。我用qlua运行它。

require "image" input_image = image.load(arg[1]) image.display{image = input_image}

如果图像很大，qt窗口只占整个屏幕，这也会拉伸图像以适应屏幕。

我无法找到防止这种情况发生的方法。

谢谢！

Answer 1

如果图像很大，请将其调整为可以配置为＆＃34;最大高度/最大宽度＆＃34;，同时保留纵横比。

示例代码：

maxSize = 480
-- find the smaller dimension, and resize it to maxSize (while keeping aspect ratio)
local iW = input:size(3)
local iH = input:size(2)
if iW < iH then
   input = image.scale(input, maxSize, maxSize * iH / iW)
else
   input = image.scale(input, maxSize * iW / iH, maxSize)
end