给定 Lua Torch 中的图像,我如何找到它的尺寸?

Given an image in Lua Torch, how can I find its dimensions?

假设我想知道 image.lena() 的高度和宽度。我应该调用什么方法? link 到图像方法的资源会很好,因为 Torch help 命令在这种情况下不起作用。

image.lena() returns 一个 Torch 三维张量,其中第一维是通道数(RGB 图像为 3),最后一个是 resp。图像的高度(行数)和宽度(列数)。

所以你需要做的就是使用如下size(dim)方法:

require 'image'

local img = image.lena()
print(torch.typename(img)) -- torch.DoubleTensor

local nchan, height, width = img:size(1), img:size(2), img:size(3)
print('nb. channels: ' .. nchan) -- 3
print('width: ' .. width .. ', height: ' .. height) -- 512, 512