Toybrick

请问多久会支持pytorch框架?据说onnx要散伙了

granis

注册会员

积分
73
楼主
发表于 2019-3-12 18:36:32 | 显示全部楼层
jefferyzhang 发表于 2019-3-7 09:24
确切消息,已经在支持中了,下一个版本应该会发布

pytorch过来onnx转换的时候反卷积和Clip(0,6)都有问题。
能给支持么?
回复

使用道具 举报

granis

注册会员

积分
73
沙发
发表于 2019-3-14 09:14:49 | 显示全部楼层
jefferyzhang 发表于 2019-3-14 09:02
反卷积支持,
Clip暂时不支持

帮我看看这个,在反卷积的时候出错。Clip是MobileNetV2中用到的,能否提供支持,实现起来也是非常简单。
W verbose file path is invalid, debug info will not dump to file.
I Current ONNX Model use ir_version 3 opset_version 9
D import clients finished
I build output layer output_473
I Try match Conv op#154
I Match [['473']] [['Conv_1']] to [['convolution_1']]
I Try match Relu op#153
I Match [['472']] [['Relu_1']] to [['relu_1']]
I Try match BatchNormalization op#152
I Match [['471']] [['BatchNormalization']] to [['batchnormalize_1']]
I Try match ConvTranspose op#151
I Match [['470']] [['ConvTranspose_1']] to [['deconvolution_1']]
E Catch exception when loading onnx model: ./model.onnx!
T Traceback (most recent call last):
T   File "rknn/api/rknn_base.py", line 321, in rknn.api.rknn_base.RKNNBase.load_onnx
T   File "rknn/base/rknnlib/converter/convert_onnx.py", line 555, in rknn.base.rknnlib.converter.convert_onnx.convert_onnx.match_paragraph_and_param
T   File "rknn/base/rknnlib/converter/convert_onnx.py", line 355, in rknn.base.rknnlib.converter.convert_onnx.convert_onnx._onnx_build_acu_layer
T   File "rknn/base/rknnlib/converter/convert_onnx.py", line 322, in rknn.base.rknnlib.converter.convert_onnx.convert_onnx._onnx_acu_param_assign
T   File "rknn/base/rknnlib/converter/convert_onnx.py", line 302, in rknn.base.rknnlib.converter.convert_onnx.convert_onnx._onnx_parse_param
T   File "rknn/base/rknnlib/converter/paragraph.py", line 21, in rknn.base.rknnlib.converter.paragraph.lambda4
T TypeError: 'int' object is not iterable
回复

使用道具 举报

granis

注册会员

积分
73
板凳
发表于 2019-3-14 10:52:40 | 显示全部楼层
jefferyzhang 发表于 2019-3-14 09:25
这个log看是匹配反卷积层时候出现问题,并不是op不支持。
先测试下你的模型转成onnx后是否可以正常推理, ...

我们用这个模型转了Mace和NCNN都没有问题。
回复

使用道具 举报

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

本版积分规则

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


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