我已经搜索过,但找不到与之相关的任何内容,因此很抱歉,如果这是一个重复的问题。

我目前正在学习OpenCV,在尝试获取laplacian图像时遇到了数组大小的问题。基本上,从cv2.pyrUp()返回的结果比从高斯金字塔得到的结果多一个“行”,这导致它抛出输入参数错误的大小。以前有没有人遇到过这个问题,该怎么解决?

这是我的初始代码:

img = cv2.imread('test.jpg', 1)
G = img.copy()
gpA =[G]
for i in range(0,5):
  G = cv2.pyrDown(G)
  gpA.append(G)

lpA = [gpA[5]]
for i in xrange(5,0,-1):
    GE = cv2.pyrUp(gpA[i])
    w,d,h = GE.shape        #debug
    print w,d,h

    x,y,z = gpA[i-1].shape  #debug
    print x,y,z

    L = cv2.subtract(gpA[i-1], GE)
    lpA.append(L)

我使用了以下代码:http://docs.opencv.org/3.1.0/dc/dff/tutorial_py_pyramids.html#gsc.tab=0

我得到输入参数错误的大小,调试结果是
58 90 3
57 90 3

但是,通过更改从pyrUp部分开始的代码以从第一级进行迭代:
lpA = [gpA[5]]
for i in xrange(0,5):
 GE = cv2.pyrUp(gpA[i+1])
    w,d,h = GE.shape          #debug
    print w,d,h
    x,y,z = gpA[i].shape      #debug
    print x,y,z
    L = cv2.subtract(gpA[i], GE)
    lpA.append(L)

我从调试中得到的结果是:
900 1440 3
900 1440 3
450 720 3
450 720 3
226 360 3
225 360 3

因此,在这种情况下,循环在遇到相同问题之前设法进行了两次迭代。

最佳答案

这里的问题是,在某一时刻您得到的图像行数不均匀。假设您有2 * n + 1行。 pyrDown将为您提供带有n + 1行的图像。因此,当您执行pyrUp时,最终将得到一幅包含2*n + 2行的图像,比以前多一幅。

解决方案是,当您执行pyrUp时,您应该这样做:

size = (gpA[i].shape[1], gpA[i].shape[0])
GE = cv2.pyrUp(gpA[i+1], dstsize = size)

这样,您的图像GE将具有与gpA[i]相同的大小。

10-06 11:14