DB icon indicating copy to clipboard operation
DB copied to clipboard

.onnx

Open sssimpleboy opened this issue 2 years ago • 17 comments

hello! when i try to transform .pth to .onnx ,an error appears: image I find that i have to remove "dataparallel(model)". image

After that ,i run again image another error appears: image it didn't produce a .onnx in my directory and there were no bugs to remind me where the problem is. So could you help me ? thank you!

sssimpleboy avatar Jun 28 '22 10:06 sssimpleboy

just now I found convert_to_onnx.py that you provided in May,but failed. image How to convert.....?

sssimpleboy avatar Jun 29 '22 05:06 sssimpleboy

请问您知道原因吗? 我在这卡了好几天啦。。。没有报错提示,啥也没有。 束手无策。哈哈哈

sssimpleboy avatar Jun 29 '22 05:06 sssimpleboy

你是怎么走到这一步的,我停在了 key error “visualize”。看代码又个cmd 参数,默认初始化为空字典。获取这个key的时候出错了。

BeanSprouts avatar Jul 07 '22 07:07 BeanSprouts

你是怎么走到这一步的,我停在了 key error “visualize”。看代码又个cmd 参数,默认初始化为空字典。获取这个key的时候出错了。

我没出现你这个错误过。。不知道耶。

sssimpleboy avatar Jul 07 '22 07:07 sssimpleboy

看你上面贴的代码,和作者提供的代码有点出入。你是做了改动吗

BeanSprouts avatar Jul 07 '22 08:07 BeanSprouts

看你上面贴的代码,和作者提供的代码有点出入。你是做了改动吗

对。

sssimpleboy avatar Jul 07 '22 08:07 sssimpleboy

有没有一种可能,这个工程的模型目前没法转成onnx。 改了opsetversion=9 报错:Couldn't export Python operator ModulatedDeformConvFunction

BeanSprouts avatar Jul 07 '22 08:07 BeanSprouts

有没有一种可能,这个工程的模型目前没法转成onnx。 改了opsetversion=9 报错:Couldn't export Python operator ModulatedDeformConvFunction 但是别人可以成功呀。。。

sssimpleboy avatar Jul 07 '22 08:07 sssimpleboy

和pytorch版本有没有关系呢,我的是torch==1.10.2+cu113 。opserversion=11,报错:Segmentation fault

BeanSprouts avatar Jul 07 '22 08:07 BeanSprouts

我1.3.0 op也是11

sssimpleboy avatar Jul 07 '22 08:07 sssimpleboy

我看了另外一个issue,里面提到 non-DCN。我的是带DCN的。我现在重新训练。

BeanSprouts avatar Jul 07 '22 08:07 BeanSprouts

对 带DCN的不行

sssimpleboy avatar Jul 07 '22 08:07 sssimpleboy

用作者提供的脚本,转不带DCN的成功了,opset_version=9

BeanSprouts avatar Jul 07 '22 09:07 BeanSprouts

用作者提供的脚本,转不带DCN的成功了,opserversion=9

你测试一下

sssimpleboy avatar Jul 07 '22 09:07 sssimpleboy

能执行推理,但是结果还没解析出来。没看到有现成用onnx推理的代码。明天调一调。不过现在感觉不太好。

BeanSprouts avatar Jul 07 '22 10:07 BeanSprouts