如何使用 criterion 来计算 Torch/Lua 中的 earth movers 距离(或如何只获得比较结果)
2017-7-5 12:14:21
收藏:0
阅读:120
评论:1
我正在尝试计算 Torch7 中两个直方图之间的距离,为此我想使用 earth mover's distance。现在我知道在 Python 中使用 https://github.com/garydoranjr/pyemd 类似的东西并不难,但我的数据在 Torch 中,需要多次执行这个计算。因此,将整个数据在 Torch7 和 Python 之间移动不是一个选项。
那么我的问题是,在 Torch7 中,哪个是最快的 earth mover 距离计算器?我已经搜索过了,但没有找到任何类似于库的东西,希望有一种更好的方法来实现这个过程,而不是像逐行翻译 Python 代码一样。尤其是 Torch 通常更擅长处理 GPU 上的事情。
编辑我已经发现了 这个,但不确定如何使用它。
我目前有以下代码:
function ColourCompareHistEMD (imagers)
sumdistance=0
k={}
for i=1,$images do
k[i]=torch.bhistc(images[i],20,-100,100)
end
for i=1,$images do
for j=1,$images do
#在这里做什么?
end
end
end
我目前的最佳猜测是这样的:
function ColourCompareHistEMD (images)
sumdistance=0
r={}
for i=1,#images do
print(images[i])
r[i]=torch.histc(images[i][1]:view(images[i][1]:nElement()),20,-100,100)
end
for i=1,#images do
for j=1,#images do
criterion = nn.EMDCriterion()
criterion:forward(r[i],r[j])
sumdistance=sumdistance+criterion.loss
end
end
return sumdistance
end
但这似乎不起作用,因为 criterion.loss 不起作用并且给我一个错误
/home/thijser/torch/install/bin/luajit: bad argument #2 to '?' (out of range at /home/thijser/torch/pkg/torch/generic/Tensor.c:704)
stack traceback:
[C]: at 0x7f2048fdc530
[C]: in function '__newindex'
/home/thijser/torch/install/share/lua/5.1/EMDCriterion.lua:52: in function 'preprocess'
/home/thijser/torch/install/share/lua/5.1/EMDCriterion.lua:255: in function 'forward'
imageSelector.lua:343: in function 'evalHueImages'
imageSelector.lua:66: in function 'evaluate'
imageSelector.lua:81: in function 'SelectTop'
imageSelector.lua:151: in function 'evolve'
imageSelector.lua:158: in function <imageSelector.lua:156>
[C]: in function 'dofile'
...jser/torch/install/lib/luarocks/rocks/trepl/scm-1/bin/th:150: in main chunk
[C]: at 0x5641c3f40470
但不确定如何使用它,例如在评论中计算图像 i 和 j 之间的 earth mover 距离。
点赞
评论区的留言会收到邮件通知哦~
推荐文章
- Lua 虚拟机加密load(string.dump(function)) 后执行失败问题如何解决
- 我想创建一个 Nginx 规则,禁止访问
- 如何将两个不同的lua文件合成一个 东西有点长 大佬请耐心看完 我是小白研究几天了都没搞定
- 如何在roblox studio中1:1导入真实世界的地形?
- 求解,lua_resume的第二次调用继续执行协程问题。
- 【上海普陀区】内向猫网络招募【Skynet游戏框架Lua后端程序员】
- SF爱好求教:如何用lua实现游戏内调用数据库函数实现账号密码注册?
- Lua实现网站后台开发
- LUA错误显式返回,社区常见的规约是怎么样的
- lua5.3下载库失败
- 请问如何实现文本框内容和某个网页搜索框内容连接,并把网页输出来的结果反馈到另外一个文本框上
- lua lanes多线程使用
- 一个kv数据库
- openresty 有没有比较轻量的 docker 镜像
- 想问一下,有大佬用过luacurl吗
- 在Lua执行过程中使用Load函数出现问题
- 为什么 neovim 里没有显示一些特殊字符?
- Lua比较两个表的值(不考虑键的顺序)
- 有个lua简单的项目,外包,有意者加微信 liuheng600456详谈,最好在成都
- 如何在 Visual Studio 2022 中运行 Lua 代码?

似乎
EMDCriterion要求输入和目标至少为二维。它还要求您比较中的点横向排列。由于torch.histc的结果是一维的,因此您可以将其重塑为二维行张量,如下所示:for i=1,#images do print(images[i]) local hist = torch.histc(images[i][1]:view(images[i][1]:nElement()),20,-100,100) r[i] = hist:reshape(1,hist:nElement()) end此外,我尝试运行代码
criterion:forward(r[i],r[j]) print(criterion.loss)结果是
nil。改为以下代码以累积损失:local loss = criterion:forward(r[i],r[j]) sumdistance = sumdistance + loss此外,如果您在嵌套的
for-loop 外定义分类器criterion = nn.EMDCriterion(),效率会更高一些。