PyQt使用QPixmap显示掩膜、截图时shape解包值不足报错如何解决

报错原因 #

OpenCV中图像矩阵的shape属性返回维度和图像通道数直接相关：三通道彩色图像会返回(高度, 宽度, 通道数)三个值，单通道二值/灰度图像仅返回(高度, 宽度)两个值。
两处报错的具体触发逻辑：

• 掩膜图像场景：cv2.inRange、cv2.bitwise_not输出的掩膜本身就是单通道二值图像，mask_frame.shape只有两个返回值，强行解包为三个变量就会触发报错
• 屏幕截图场景：image_grab_frame被处理为单通道灰度图像，同样不存在通道数维度
  此外原有代码还存在隐藏问题：单通道图像直接用QImage.Format_RGB888格式转换会出现颜色显示异常，需要同步修复。

修复方案 #

核心处理逻辑：先判断图像通道数，单通道图像先通过cv2.cvtColor(单通道图像, cv2.COLOR_GRAY2BGR)转为三通道图像，再执行后续解包、转换操作。同时给QImage构造补充步长参数，避免行对齐导致的显示花屏问题。

1. 掩膜图像功能修复后代码 #

mask = cv2.inRange(hsv, lower_hue, upper_hue)
mask1=cv2.bitwise_not(mask)
hsv_height, hsv_width, hsv_channel = hsv_image.shape
hsv_step = hsv_channel * hsv_width

# 新增通道数判断与转换逻辑
if len(mask_frame.shape) == 2:
    mask_frame = cv2.cvtColor(mask_frame, cv2.COLOR_GRAY2BGR)
mask_height, mask_width, mask_channel = mask_frame.shape
mask_step = mask_channel * mask_width

# 补充步长参数mask_step
convertToQFormat = QImage(mask_frame.data, mask_frame.shape[1], mask_frame.shape[0], mask_step, QImage.Format_RGB888)
pic = convertToQFormat.scaled(1280, 720, Qt.KeepAspectRatio)
self.normal_screen.setPixmap(QPixmap.fromImage(pic))

2. 屏幕截图功能修复后代码 #

# 新增通道数判断与转换逻辑
if len(image_grab_frame.shape) == 2:
    image_grab_frame = cv2.cvtColor(image_grab_frame, cv2.COLOR_GRAY2BGR)
screen_grab_height, screen_grab_width, screen_grab_channel = image_grab_frame.shape
screen_grab_step = screen_grab_channel * screen_grab_width

#Display image grab#
# 补充步长参数screen_grab_step
convertToQFormat = QImage(image_grab_frame.data,image_grab_frame.shape[1], image_grab_frame.shape[0], screen_grab_step, QImage.Format_RGB888)
image_grab_pic = convertToQFormat.scaled(1280, 720, Qt.KeepAspectRatio)     

self.normal_screen.setPixmap(QPixmap.fromImage(image_grab_pic))

可选优化 #

OpenCV默认读取的图像为BGR格式，QImage.Format_RGB888为RGB格式，如果出现颜色显示偏色，可以在转QImage前加一行frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)调整颜色通道顺序。

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