Matplotlib报TypeError: Invalid dimensions for image data 求解

首先得指出你代码里的一个小问题：你定义的jpg变量写法不对，'b'\xff...''是带字符串包裹的字节字面量，实际应该直接写成字节对象b'\xff\xd8...'，不然后续处理会出问题。

你遇到的报错TypeError: Invalid dimensions for image data，本质原因是：np.fromstring（现在更推荐用np.frombuffer）得到的是一维的uint8数组，而matplotlib的imshow需要的是二维数组（灰度图，形状是(height, width)）或者三维数组（彩色图，形状是(height, width, 3)），直接传一维数组它根本没法识别成图像结构。

而且你提到想用OpenCV处理，那正确的流程应该是先把JPEG字节码解码成OpenCV能识别的图像矩阵，再进行显示或其他处理。下面给你两种可行的解决方案：

方案一：用OpenCV直接解码并显示 #

这是最贴合你需求的方式，OpenCV自带的imdecode函数可以直接把字节数组解码成图像：

import numpy as np
import cv2

# 修正后的JPEG字节对象
jpg = b'\xff\xd8\xff\xe0\x00\x10JFIF\x00\x01\x01\x01\x00`\x00`\x00\x00\xff\xfe\x00>CREATOR: gd-jpeg v1.0 (using IJG JPEG v80), default quality\n\xff\xdb\x00C\x00\x08\x06\x06\x07'

# 将字节码转为numpy一维数组
nparr = np.frombuffer(jpg, np.uint8)
# 解码成图像矩阵：IMREAD_COLOR表示彩色图，IMREAD_GRAYSCALE表示灰度图
img = cv2.imdecode(nparr, cv2.IMREAD_COLOR)

# 检查解码是否成功（避免字节码不完整导致的错误）
if img is not None:
    cv2.imshow('Processed Image', img)
    cv2.waitKey(0)  # 按任意键关闭窗口
    cv2.destroyAllWindows()
else:
    print("图像解码失败，请检查JPEG字节码是否完整")

方案二：用matplotlib显示（配合解码步骤） #

如果你还是想用matplotlib来显示，需要先把字节码解码成正确维度的图像数组，注意OpenCV默认是BGR格式，matplotlib是RGB格式，需要转换一下：

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

jpg = b'\xff\xd8\xff\xe0\x00\x10JFIF\x00\x01\x01\x01\x00`\x00`\x00\x00\xff\xfe\x00>CREATOR: gd-jpeg v1.0 (using IJG JPEG v80), default quality\n\xff\xdb\x00C\x00\x08\x06\x06\x07'

nparr = np.frombuffer(jpg, np.uint8)
img_bgr = cv2.imdecode(nparr, cv2.IMREAD_COLOR)
# 转换BGR到RGB格式
img_rgb = cv2.cvtColor(img_bgr, cv2.COLOR_BGR2RGB)

plt.imshow(img_rgb)
plt.axis('off')  # 可选，隐藏坐标轴
plt.show()

或者用PIL库来解码（不需要格式转换，PIL默认是RGB）：

import numpy as np
from matplotlib import pyplot as plt
from PIL import Image
import io

jpg = b'\xff\xd8\xff\xe0\x00\x10JFIF\x00\x01\x01\x01\x00`\x00`\x00\x00\xff\xfe\x00>CREATOR: gd-jpeg v1.0 (using IJG JPEG v80), default quality\n\xff\xdb\x00C\x00\x08\x06\x06\x07'

# 用BytesIO包装字节码，让PIL可以读取
img = Image.open(io.BytesIO(jpg))
# 转为numpy数组
img_np = np.array(img)

plt.imshow(img_np)
plt.axis('off')
plt.show()

总结一下：永远不要直接把原始的JPEG字节数组传给图像显示函数，必须先通过解码工具（OpenCV的imdecode、PIL的Image.open）把字节码转换成符合图像维度要求的矩阵，这样才能正常显示和处理。

内容的提问来源于stack exchange，提问作者user1592380