python - Onnx 模型输入大小与 Opencv 帧大小不同
问题描述
如何将 OpenCV 框架转换为适合我的 ONNX 模型接受的尺寸?目前,我的 ONNX 模型输入形状是 [32, 3, 256, 224] 但是当我使用 OPENCV 调整大小并打印 img 形状时,它是 (224,256, 3)。
解决方案
分离通道,转置图像尺寸并将通道放在一起。您可以使用这一行代码:
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)
推荐阅读
- python - os 模块 python 正在工作,但输出不正确
- regex - 如何为文本字段实现正则表达式以在 Extjs 中一次允许一个逗号一个点
- laravel - Laravel - 试图获取非对象的属性“名称”
- java - Spring SAML:后退浏览器按钮导致验证 SAML 消息时出错
- ios - React Native 0.60.4 从 0.59.9 升级
- python - 在 TfidfVectorizer 中删除法语和英语中的停用词
- c# - ASP.NET Core Identity,如何覆盖 IdentityRole Name 唯一索引?我想存储关于租户的重复名称
- web-scraping - 基于页面内容的操作
- javascript - 为什么我的选择器无法读取“状态”?状态未定义
- c# - 将列表项从任务添加到组合框