我已经搜索过,但找不到与之相关的任何内容,因此很抱歉,如果这是一个重复的问题。
我目前正在学习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]
相同的大小。