Python OpenCV框架形状不可用

我有一个CNN，它需要输入图像的形状[通道，大小，大小]，但当我使用imutils视频流功能获得帧时，我得到的是形状[大小，大小，通道]的帧

使用numpy REFORMATE重塑此帧会使我的CNN无法识别帧中的内容，并且无法正确分类任何内容

有没有办法指定我想要的框架的形状？或者以这样一种方式重塑框架，以保留图像

from imutils.video import VideoStream
import time
vs = VideoStream(src=1).start()
time.sleep(2.0)
frame = vs.read()
print(frame.shape)  # output: (480, 640, 3) -> I need (3, 480, 640)

---

我认为您正在寻找的函数是转置。假设

帧

为 对于numpy阵列，您可以使用：

frame_cnn = frame.transpose([2,0,1])