ncnn icon indicating copy to clipboard operation
ncnn copied to clipboard

Published 20 hours ago • verified publisher icon Tencent

想请教下模型输入如果维度大于3时的情况

Open johninn opened this issue 2 years ago • 1 comments

假如我的模型pytorch下输入是nchw，读取图片后用numpy.expand_dims做升维。 转化为ncnn的模型后，ncnn::mat好像只支持三维。直接输入三维mat不会报错，但是模型输出结果和pytorch的不一样。 那是需要对mat做升维操作还是需要把模型输入层做修改？

johninn avatar Jun 08 '22 03:06 johninn

同问，有什么解决方法吗？ lol

s1ghhh avatar Jul 02 '22 15:07 s1ghhh