lua - 给定 Lua Torch 中的图像，我如何找到它的尺寸？

Question

假设我想知道image.lena(). 我应该调用什么方法？help由于 Torch命令在这种情况下不起作用，因此指向图像方法资源的链接会很棒。

Answer 1

image.lena()返回一个 Torch 3 维张量，其中第一维是通道数（RGB 图像为 3），最后一个是通道数。图像的高度（行数）和宽度（列数）。

因此，您需要做的就是使用以下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