如何使用opencv计算2个numpy数组（即“直方图”）的“ EMD”？

Z时代
2024-01-10
分类：问答

由于我是opencv的新手，所以我不知道如何将cv.CalcEMD2函数与numpy数组一起使用。

我有两个数组：

a=[1,2,3,4,5]  
b=[1,2,3,4]

我怎么能转移numpy array到CVhistogram从Cvhistogram该函数的参数signature？

我希望回答问题的任何人opencv通过提供的解决方案来解释所有使用的功能。

“ EMD” ==推土机的距离。

更新：-

同样，如果有人可以告诉我如何设置cv.CalcEMD2参数（即"signature"使用numpy数组），这将很有帮助！

注意：-

*对于可能对此问题感兴趣的人，此答案需要更多测试。

回答：

您必须根据权重和坐标定义数组。如果您有两个表示一维直方图的数组a = [1,1,0,0,1]和b = [0,1,0,1]，那么numpy数组应如下所示：

a = [[1 1]
     [1 2]
     [0 3]
     [0 4]
     [1 5]]
b = [[0 1]
     [1 2]
     [0 3]
     [1 4]]

请注意，行数可以不同。列数应为尺寸+1。第一列包含权重，第二列包含坐标。

下一步是在将numpy数组输入为CalcEMD2函数的签名之前，将数组转换为CV_32FC1 Mat。代码如下所示：

from cv2 import *
import numpy as np
# Initialize a and b numpy arrays with coordinates and weights
a = np.zeros((5,2))
for i in range(0,5):
    a[i][1] = i+1
a[0][0] = 1
a[1][0] = 1
a[2][0] = 0
a[3][0] = 0
a[4][0] = 1
b = np.zeros((4,2))
for i in range(0,4):
    b[i][1] = i+1
b[0][0] = 0
b[1][0] = 1
b[2][0] = 0
b[3][0] = 1
# Convert from numpy array to CV_32FC1 Mat
a64 = cv.fromarray(a)
a32 = cv.CreateMat(a64.rows, a64.cols, cv.CV_32FC1)
cv.Convert(a64, a32)
b64 = cv.fromarray(b)
b32 = cv.CreateMat(b64.rows, b64.cols, cv.CV_32FC1)
cv.Convert(b64, b32)
# Calculate Earth Mover's
print cv.CalcEMD2(a32,b32,cv.CV_DIST_L2)
# Wait for key
cv.WaitKey(0)