Toybrick

为什么 rknn.inference 模型输出的数组维度与原生模型不同

程子

中级会员

积分
386
楼主
发表于 2019-3-20 21:40:03 | 显示全部楼层
RKNN的维度格式是 BCHW (Batch, Channel, Height, Width),并且输出是摊平的。
  1. output = rknn.inference(inputs=[input])
  2. output = np.transpose(output.reshape((batch, channel, height, width)), (0, 2, 3, 1))
复制代码
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

产品中心 购买渠道 开源社区 Wiki教程 资料下载 关于Toybrick


快速回复 返回顶部 返回列表