python - 将 opencv 阈值应用于 numpy 数组

Question

我正在尝试将 opencv 的 Threshold 函数应用于 numpy 数组。我正在使用 opencv 2.1 的 python 绑定(bind)。它是这样的:

import cv
import numpy as np
a = np.random.rand(1024,768)
cv.Threshold(a,a,0.5,1,cv.CV_THRESH_BINARY)

这会引发错误:

OpenCV Error: Unsupported format or combination of formats () in threshold

所以，我不相信我知道我在做什么，但我希望 Threshold 会像 Smooth 一样工作，我可以在其中运行

cv.Smooth(a,a)

没有问题，最终得到一个平滑的(呃)图像。我不确定如何考虑 numpy 数组的“格式”，因为 opencv 看到了它们，但如果不需要的话，我不愿意将 numpy 数组转换为 opencv 图像格式(而且我所有的尝试都失败了所以无论如何)。

我想知道为什么 Threshold 没有以我试图让它工作的明显天真的方式工作，如果知道我应该做什么就太好了。

附言我知道我可以自己对 numpy 数组执行阈值操作，但我正在尝试找出 opencv。

Answer 1

显然 Threshold 方法比 Smooth 更挑剔 - 它只适用于 8 位整数/32 位 float 组(参见 here )所以你上面的代码片段不会工作，因为 numpy数组默认为 float64。

因此，如果您更改创建数组的行以将精度强制为 32 位 float

>>> a = np.array(np.random.rand(1024,768),dtype=‘float32’)

然后很高兴达到阈值:

>>> ((a>0) & (a<1)).sum()
786432
>>> cv.Threshold(a,a,0.5,1,cv.CV_THRESH_BINARY)
>>> ((a>0) & (a<1)).sum()
0