Onnx 模型输入大小与 Opencv 帧大小不同

【问题标题】：Onnx model input size different from Opencv frame sizeOnnx 模型输入大小与 Opencv 帧大小不同
【发布时间】：2020-07-18 16:49:47
【问题描述】：

如何将 OpenCV 帧转换为适合我的 ONNX 模型接受的尺寸？目前，我的 ONNX 模型输入形状是 [32, 3, 256, 224]，但是当我使用 OPENCV 调整大小并打印 img 形状时，它是 (224,256, 3)。

【问题讨论】：

标签： python numpy opencv onnx

【解决方案1】：

分离通道，转置图像尺寸并将通道放在一起。您可以使用这一行代码：

np.array([np.transpose(img[:, :, 0]), np.transpose(img[:, :, 1]), np.transpose(img[:, :, 2])])

例子：

a = np.zeros((224,256, 3))
b = np.array([np.transpose(a[:, :, 0]), np.transpose(a[:, :, 1]), np.transpose(a[:, :, 2])])
b.shape #returns (3, 256, 224)

【讨论】：

谢谢。我一开始就在尝试类似的事情。我最终找到了一种更简单的方法。