Toybrick

模型的inference输出维度问题

jay-fit-future

新手上路

积分
38
发表于 2019-5-28 19:05:06    查看: 1316|回复: 1 | [复制链接]    打印 | 显示全部楼层
本帖最后由 jay-fit-future 于 2019-5-30 11:07 编辑

Hi,向各位咨询和求助一个问题,

我现在在rk3399pro的docker环境下部署了自己的模型,我的模型输出是一个3维张量,在电脑上用gpu跑出来是一个(46,46,57)的输出。用同样的模型,我在3399pro的模拟环境下跑,输出的结果是一个(1,120612)的输出(120612 = 46*46*57)。我试着对输出结果进行reshape,无论是(46,46,57)还是(57,46,46)亦或是(46,57,46),结果都不太正确.
回复

使用道具 举报

jay-fit-future

新手上路

积分
38
 楼主| 发表于 2019-5-30 11:09:43 | 显示全部楼层
参考了直播的解释,已经解决。 结果不正确是我自己在处理输入的时候的问题。针对输出维度,我用的输入是NCHW,所以对应的输出reshape到NCHW就好了。 在我这里我reshape到了(1,57,46,46)。
回复

使用道具 举报

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

本版积分规则

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


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