python - Shape Context Error 通过计算两个不同形状的距离

Question

简而言之，我想通过“形状上下文”匹配来比较不同的道路标记。

我的第一个问题你可以在下面看到:Matching shapes of road marking from OpenData

我解决了我的第一个问题，但现在出现了一个新错误。这是我的代码:

import cv2
import numpy as np

# read data
datapath = "/Users/output/test/";
a = cv2.imread(datapath+"template_orig.png");
b = cv2.imread(datapath+"template.png");

imgray_a = cv2.cvtColor(a,cv2.COLOR_BGR2GRAY)
ret_a,thresh_a = cv2.threshold(imgray_a,127,255,0)

imgray_b = cv2.cvtColor(b,cv2.COLOR_BGR2GRAY)
ret_b,thresh_b = cv2.threshold(imgray_b,127,255,0)


# find contours
_, ca, _ = cv2.findContours(thresh_a, cv2.RETR_CCOMP, cv2.CHAIN_APPROX_SIMPLE)
_, cb, _ = cv2.findContours(thresh_b, cv2.RETR_CCOMP, cv2.CHAIN_APPROX_SIMPLE)
print(np.shape(ca[0]), np.shape(cb[0]))

# generate distance --> Hausdorff OR ShapeContext
hd = cv2.createHausdorffDistanceExtractor()
sd = cv2.createShapeContextDistanceExtractor()

d1 = hd.computeDistance(ca[0],cb[0])
d2 = sd.computeDistance(ca[0],cb[0])


print(d1, " ", d2)

当我比较a时(原来的转弯箭头) 用b(提取的转向箭头)没有问题，但是当我比较a 和c(任何其他测试“形状匹配”算法)出现以下错误:

OpenCV Error: Assertion failed (type == CV_64FC2) in gemmImpl, file /Users/travis/build/skvark/opencv-python/opencv/modules/core/src/matmul.cpp, line 1218

Traceback (most recent call last): File "/test_shape.py", line 74, in d2 = sd.computeDistance(ca[0],cb[0])

cv2.error: /Users/travis/build/skvark/opencv-python/opencv/modules/core/src/matmul.cpp:1218: error: (-215) type == CV_64FC2 in function gemmImpl

只有“shape context”的距离生成函数才会出错，“Hausdorff”的距离生成函数不会出错

Answer 1

好的，我认为问题是，a 和 b 或 a 和 c 不具有相同的像素高度(a:131 x 32px，b/c:29 x 18 px)。当我将 b 或 c 的图像大小时更改为更高分辨率(如 131 x 81 像素)时，错误消失并且正在计算“形状上下文”的距离。