我有一个由点[[x1,y1],[x2,y2],...,[xn,yn]]数组定义的形状,以及一个图像(img1),其中(几乎)是I需要找到该形状的位置,这意味着如果在图像(img2)上的任意位置绘制此形状,我会发现仿射变换从img1变为img2。我设法做到了cv2.findTransformECC。我得到了warp_matrix。
[img1] https://i.imgur.com/097V8YM.png [img2] https://i.imgur.com/dNUrgE8.png
代码:
def get_gradient(im) :
# Calculate the x and y gradients using Sobel operator
grad_x = cv2.Sobel(im,cv2.CV_32F,1,0,ksize=3)
grad_y = cv2.Sobel(im,cv2.CV_32F,0,1,ksize=3)
# Combine the two gradients
grad = cv2.addWeighted(np.absolute(grad_x), 0.5, np.absolute(grad_y), 0.5, 0)
return grad
img1=cv2.imread('img1.png',0)
points=np.array([[ 834, 429],
[ 867, 419],
[ 900, 409],
[ 934, 400],
[ 967, 391],
[1001, 382],
[1035, 375],
[1069, 369],
[1102, 364],
[1136, 361],
[1170, 361],
[1204, 362],
[1238, 365],
[1272, 370],
[1306, 376],
[1340, 385]])
img2=np.zeros_like(img1)
cv2.polylines(img2,[points],False,255,4)
warp_mode = cv2.MOTION_AFFINE
warp_matrix = np.eye(2, 3, dtype=np.float32)
criteria = (cv2.TERM_CRITERIA_EPS | cv2.TERM_CRITERIA_COUNT, 3000, 1e-5)
(cc, warp_matrix) = cv2.findTransformECC (get_gradient(img1), get_gradient(img2),warp_matrix, warp_mode, criteria)
img3 = cv2.warpAffine(img2, warp_matrix, (img1.shape[1],img1.shape[0]), flags=cv2.INTER_LINEAR + cv2.WARP_INVERSE_MAP)
结果: [img3] https://i.imgur.com/zCXJoyJ.png
然后我想用cv2.transform在合适的位置上直接绘制形状,但是我的工作很奇怪,就好像旋转角度与不良符号一起使用一样。
以下代码是我的问题,请参见img4中的结果:
warp_points=cv2.transform(np.reshape(points,(points.shape[0],1,2)),warp_matrix)
img4=img1.copy()
cv2.polylines(img4,[warp_points],False,100,4) #100 : gray
[img4] https://i.imgur.com/JvUHBVK.png
通过提前,(对于任何英语错误,对不起,这不是我的mother堂)
答案 0 :(得分:0)
尝试在转换图像时使用的cv2.findTransformECC函数中交换图像的顺序,并使用新的扭曲矩阵转换点。在浏览opencv文档后,也请尝试更改与cv2.findTransformECC的“ criteria”属性关联的值。我确信这些会有所帮助,因为我过去曾遇到过类似的问题。