python - 形状不可用的 OpenCV 框架

Question

我有一个 CNN，它需要输入图像的形状为 [ channel 、大小、大小]，但是当我使用 imutils videostream 函数获取帧时，我得到的帧的形状为 [大小、大小、 channel ]。

使用 numpy reshape reshape 此框架使我的 CNN 无法识别框架中的内容，并且无法正确分类任何内容。

有什么方法可以指定我想要框架的形状吗？或者以我保留图像的方式 reshape 框架？

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)

Answer 1

我认为您正在寻找的功能是转置。假设 frame是
一个 numpy 数组，您可以使用:

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