Question

我想要使用pcolormesh显示一个图像，但我真的不明白这应该如何正常工作。我有一个特定颜色的X和相应的Y，但是如果我在pcolormesh中输入一个普通数组作为C，我会收到一个错误。

我的代码：

# load image
img = cv2.imread('Distorted_resized_50.jpg')
img_array = np.asarray(img)
height, width, channels = img.shape
gray_img = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)

# create vector matrix
U, V = np.meshgrid(range(gray_img.shape[1]),
                   range(gray_img.shape[0]))
UV = np.vstack((U.flatten(),
                V.flatten())).T

H, mask = cv2.findHomography(UV_cp, XYZ_gcp)
UV_warped = cv2.perspectiveTransform(np.array([UV]).astype(np.float32), H)
UV_warped = UV_warped[0]
UV_warped = UV_warped.astype(np.int)
X_warped = UV_warped[:,0].reshape((height, width))
Y_warped = UV_warped[:,1].reshape((height, width))

fig, axs = plt.subplots(figsize=(15,10))
axs.pcolormesh(X_warped, Y_warped, img_array)

有谁可以帮助我？网站上的解释对我来说不是很清楚。一切正常，直到fig, axs = plt.subplots(figsize=(15,10))

回溯：

Traceback (most recent call last):
  File "C:\Users\Yorian\Desktop\TU\Stage Shore\python_files\Rectificatie dmv foto thuis\rectify.py", line 53, in <module>
    ax.pcolormesh(X_warped, Y_warped, img_array)
  File "C:\Python27\lib\site-packages\matplotlib\axes.py", line 7734, in pcolormesh
X, Y, C = self._pcolorargs('pcolormesh', *args, allmatch=allmatch)
  File "C:\Python27\lib\site-packages\matplotlib\axes.py", line 7350, in _pcolorargs
numRows, numCols = C.shape
ValueError: too many values to unpack

Answer 1

我最近遇到了一个类似的问题，并且在不了解更多信息的情况下，我猜测问题是您有一个xdim，由ydim by 3 array，plt.pcolormesh期望一个2d标量值数组具有r，g和b的值（介于0和255之间）。

话虽如此，您可以做一些事情：

将图像显示为灰度，首先通过 skimage.color.rgb2grey(image)并使用pcolormesh和cmap='binary'

进行绘图

或使用plt.imshow进行绘图，并按照本文中的建议使用kwarg=extent Matplotlib: how to make imshow read x,y coordinates from other numpy arrays?

matplotlib的pcolormesh抛出ValueError：解压缩的值太多了

1 个答案: