lua - 在 Lua 中将二进制数据转换为 Torch 张量

Question

我有使用 luasocket 从 url 下载图像的 Lua 代码:

local http = require('socket.http')
local image = require('image')

image_url = 'https://www.somedomain.com/someimage.jpg'
local body, code = http.request(image_url) -- body has jpg binary data
if not body then error(code) end -- check for errors

为了将此图像读入 Torch 张量，我将其保存在 jpg 文件中并使用 image.load 读取它:

-- open a file in binary mode to store the image
local f = assert(io.open('./temp.jpg', 'wb')) 
f:write(body)
f:close()

tensor = image.load('temp.jpg')

有没有一种方法可以直接将二进制 jpg 数据转换为 torch 张量，而无需从硬盘驱动器进行写入和读取？像这样的东西:

tensor = CovertBinaryDataToTorchTensor(body)

谢谢!

Answer 1

参见 image.decompressJPG .

您只需先将 body 字符串打包到 ByteTensor 中。这可以通过用 storage 构造这个张量来完成。可以用 string(str) 设置他的内容.