模型的inference输出维度问题

[复制链接] · 发表于 2019-5-28 19:05:06

本帖最后由 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），结果都不太正确.

只看该作者 · 发表于 2019-5-30 11:09:43

参考了直播的解释，已经解决。结果不正确是我自己在处理输入的时候的问题。针对输出维度，我用的输入是NCHW，所以对应的输出reshape到NCHW就好了。在我这里我reshape到了（1,57,46,46）。