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

Question

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

Answer 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)