如何计算2个numpy数组的"EMD",即使用opencv的"histogram"？

Question

由于我是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数组!!

注意: -
*对于那些可能对此问题感兴趣的人,此答案需要更多测试.

Answer 1

您必须根据权重和坐标定义数组.如果你有两个数组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)

请注意,CalcEMD2的第三个参数是欧几里德距离CV_DIST_L2.第三个参数的另一个选项是Manhattan Distance CV_DIST_L1.

我还想提一下,我编写了代码来计算地球移动器在Python中两个2D直方图的距离.你可以在这里找到这个代码.