如何在python中将2D图像重塑为3D

Question

我有以下代码：

import numpy as np
import os
import cv2

mask_red = np.load('C:/Users/kjbaili/.spyder-py3/GMM/gmm_color_object_detection- 
master/roipoly_annotate/Neuer Ordner/2.2.npy')

img = cv2.imread('C:/Users/kjbaili/.spyder-py3/GMM/gmm_color_object_detection- 
master/roipoly_annotate/Neuer Ordner/2.2.png')

pixel_red = img[mask_red]

print(pixel_red.shape)

cv2.imshow('red_pixel',pixel_red)

cv2.waitKey(0)

img的形状为（900x1200x3） 掩码_red是一个（900x1200）阵列

pixel_red是仅包含红色像素的提取roi，ans具有形状（51605,3） 我试图做的是使用imshow将像素视为一幅图像，因此需要将其重塑为（高度、宽度、通道）

请注意，当我按原样查看像素_红色时，我会得到一条无法理解的矩形线，这条线是有意义的，因为红色_像素的实际形状

因此，任何人都知道如何正确地查看图像

提前谢谢

Answer 1

要显示img的红色通道，请使用此选项（以BGR格式加载imread，这就是为什么2）

   pixel_red = img[:,:,2]