search

WenmuZhou / DBNet.pytorch

A pytorch re-implementation of Real-time Scene Text Detection with Differentiable Binarization
Apache License 2.0
943 stars 249 forks source link

mobilenetv3 转onnx遇到个错 #56

Open zzdang opened 4 years ago

zzdang commented 4 years ago

请问转onnx模型时遇到: Type Error: Type parameter (T) bound to different types (tensor(float) and tensor(int64) in node (Add_4) 大神们遇到了吗,咋解决呀

daixiangzi commented 4 years ago

我转的时候，没有出现该错误但是出现了其他错误，不过已经解决了。

daixiangzi commented 4 years ago

请问转onnx模型时遇到: Type Error: Type parameter (T) bound to different types (tensor(float) and tensor(int64) in node (Add_4) 大神们遇到了吗,咋解决呀

你可以帖下你的pytorch版本，当然，最好直接升级到1.6，这样错误会少很多。

buaaxiejun commented 3 years ago

我转的时候，没有出现该错误但是出现了其他错误，不过已经解决了。

您好，请教一下，您是怎么转成ONNX呢？可以贴一些代码吗？蟹蟹~

ZZHHogan commented 3 years ago

我转的时候，没有出现该错误但是出现了其他错误，不过已经解决了。

您好，请教一下，您是怎么转成ONNX呢？可以贴一些代码吗？蟹蟹~

你好，请问你解决了吗？我也遇到这个问题

19ethan commented 3 years ago

UserWarning: ONNX export failed on threshold because non-zero threshold not supported warnings.warn("ONNX export failed on " + op + " because " + msg + " not supported")他是说一个算子不支持，但是这个算子应该是后处理的分割算子，跟backbone没关系，我这个能转 resnet18的就应该能转 mobilev3的，还是 mobilev3的backbone里有这个算子了？

SWHL commented 2 years ago

参考这个：https://github.com/WenmuZhou/DBNet.pytorch/issues/98#issuecomment-901577100 可以成功转换