-
Notifications
You must be signed in to change notification settings - Fork 27
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
我尝试用你的提供的代码转onnx时碰到了TypeError: upsample_bilinear2d() received an invalid combination of arguments #17
Comments
可以看一下这个issue: 你这里看起来像是downsample_ratio的数据类型不对。 |
你看看upsample_bilnear2d对于scale_factor的要求吧,需要是 float元组或float,不是Tensor |
感谢,成功转成openvino了,就是不知道怎么推理测试,网上例子都是针对图片的,我要输入的是视频该怎么做呢 |
可以参考我写的一个动态onnx推理的python版本,在我fork的分支里的inference_onnx.py(可以推理视频): 以及我在RVM里的PR: 我这个仓库的C++ API有推理视频的接口啊,你看下README.md |
请看以下这个issue中的讨论,已经有说明: 至于理解为什么这样做,可能需要你看一下RVM的论文,以及它的pytorch源码进行理解。 |
这我就不太清楚了,openvino我也没玩过 |
你的数据预处理出错了。为啥是减去这些均值呢?RVM的源码没有啊。它的预处理是: transform = transforms.Compose([
transforms.Resize(input_resize[::-1]),
transforms.ToTensor()
]) transforms.ToTensor()里面做了归一化,x=x/255. |
你要不考虑直接用RVM官方提供的onnx吧,然后用我写的inference_onnx.py来推理,效果应该是没问题的。openvino我是真不了解,用的太少了。 |
老哥你有试过按静态输入转成的onnx的效果吗,就是像我上面那样,我想知道是不是就我这样,可否麻烦你测试下,磕头 |
我没有保留我转的静态onnx,我拿来转静态的MNN和TNN文件后就删掉了。ONNXRuntime版本的我是拿动态的来推的。过去大半年啦..... |
不好意思,感谢老哥了 |
鉴于问这个问题的人比较多,我写了一份详细的知乎教程,请参考: |
这是我的model的forward
这是报错
我是这样运行的
python Exporter.py --model-variant mobilenetv3 --precision float32 --opset 13 --device cpu --output D:\下载\test.onnx
The text was updated successfully, but these errors were encountered: