提示LeakyRelu操作类型不支持
在此一键设置昵称,即可参与社区互动!
发表于2020年11月12日 11:48:24
直达本楼层的链接
地板
显示全部楼层
回复:捕虫少年 发表于 2020-7-3 09:18 你好,请问问题还存在吗 您好,我也出现了同样的问题。我使用ONNX将pytorch的yolov3模型转换为了tensorflow的.pb模型。在Atlas 300平台上使用omg进行模型转换,显示“Unsupport op type LeakyRelu”。然而算子清单中有这一op。求各位大佬帮忙! pytorch版yolov3我使用的https://github.com/ultralytics/yolov3 tensorflow-gpu版本1.15.0 pytorch版本1.7.0 onnx版本1.8.0 |
|
发表于2020年11月12日 11:58:24
直达本楼层的链接
5#
显示全部楼层
回复:小猹狸 发表于 2020-11-12 11:48 您好,我也出现了同样的问题。我使用ONNX将pytorch的yolov3模型转换为了tensorflow的.pb模型。在Atlas 300平台上使用omg进行模 您好,可以发一下详细的报错截图吗,目前转onnx首批模型都还没有精度性能达标,后续会支持 |
|
发表于2020年11月13日 09:34:18
直达本楼层的链接
6#
显示全部楼层
回复:捕虫少年 发表于 2020-11-12 11:58 您好,可以发一下详细的报错截图吗,目前转onnx首批模型都还没有精度性能达标,后续会支持 昨天我在论坛里找了一下,发现可能是TensorFlow版本的原因导致的问题。但是我使用的onnx2tensorflow代码要求tensorflow版本为1.15.0,于是我就在pytorch中将网络里LeakyReLU换为了PReLU,之后就没有报这个算子不支持。 但后续又出现了一些问题。在Pytorch中使用Tensor.view操作和torch.cat操作经过onnx转换会在tensorflow的graph中分别生成SparseToDense和NoOp节点,而这两个节点用omg转换时都会报“Unsupport op type”。 我使用的网络中Tensor.view操作可以去掉,而torch.cat操作没找到替换的方法。感谢大佬帮忙! |
|