我的OpenCV重映射()有什么问题?

时间:2016-05-13 15:16:02

标签: python python-3.x opencv image-processing opencv3.0

我从https://stackoverflow.com/a/10374811/4828720的答案Image transformation in OpenCV中获取了代码,并尝试将其调整为我的图像。

我的源图片: My source image

在其中,我确定了方格砖中心的像素坐标,如下所示:

Source points

我的目标分辨率是784.我计算了像素的目标坐标。我得到的代码是:

import cv2
from scipy.interpolate import griddata
import numpy as np

source = np.array([
    [315, 15],
    [962, 18],
    [526, 213],
    [754, 215],
    [516, 434],
    [761, 433],
    [225, 701],
    [1036, 694],
], dtype=int)

destination = np.array([
     [14, 14],
     [770, 14],
     [238, 238],
     [546, 238],
     [238, 546],
     [546, 546],
     [14, 770],
     [770, 770]
], dtype=int)

source_image = cv2.imread('frames.png')

grid_x, grid_y = np.mgrid[0:783:784j, 0:783:784j]
grid_z = griddata(destination, source, (grid_x, grid_y), method='cubic')
map_x = np.append([], [ar[:,1] for ar in grid_z]).reshape(784,784)
map_y = np.append([], [ar[:,0] for ar in grid_z]).reshape(784,784)
map_x_32 = map_x.astype('float32')
map_y_32 = map_y.astype('float32')
warped_image = cv2.remap(source_image, map_x_32, map_y_32, cv2.INTER_CUBIC)
cv2.imwrite("/tmp/warped2.png", warped_image)

如果我运行它,没有任何源点最终到达预定的目的地,但我得到一个扭曲的混乱。我在这里添加了目的地点:

My result

我哪里错了?我注意到我的网格和地图数组的分布不如示例中那样好。点数太少了吗?我是否需要在常规网格中使用它们?我只尝试使用外角的四个点而没有运气。

2 个答案:

答案 0 :(得分:1)

如果您只有8个点可以扭曲图像中没有真正的失真,我建议使用here所述的透视变换。

您引用的链接会尝试消除导致非直线的其他扭曲,但图像中的所有线条都是直线。

代码如下:

import cv2
import numpy as np
import matplotlib.pyplot as plt

img = cv2.imread('image.png')
rows,cols,ch = img.shape

pts1 = np.float32([
    [315, 15],
    [962, 18],
    [225, 701],
    [1036, 694],
], dtype=int)

pts2 = np.float32([
     [14, 14],
     [770, 14],
     [14, 770],
     [770, 770]
], dtype=int)

M = cv2.getPerspectiveTransform(pts1,pts2)

dst = cv2.warpPerspective(img,M,(784,784))

plt.subplot(121),plt.imshow(img),plt.title('Input')
plt.subplot(122),plt.imshow(dst),plt.title('Output')
plt.show()

enter image description here

答案 1 :(得分:0)

整个问题是我再次被numpy的行/列索引而不是x / y弄糊涂了。 #opencv IRC频道中有人指出了这一点。我的源和目标数组必须切换其列:

source = np.array([
    [15, 315],
    [18, 962],
    [213, 526],
    [215, 754],
    [434, 516],
    [433, 761],
    [701, 225],
    [694, 1036],
], dtype=int)

destination = np.array([
     [14, 14],
     [14, 770],
     [238, 238],
     [238, 546],
     [546, 238],
     [546, 546],
     [770, 14],
     [770, 770]
], dtype=int)

然后它按预期工作(忽略丑陋的扭曲,这是一个简化的坐标列表来找到bug):

enter image description here