YOLOv5_NCNN icon indicating copy to clipboard operation
YOLOv5_NCNN copied to clipboard

ENet转换

Open Fight-hawk opened this issue 3 years ago • 7 comments

请教一个enet是如何转换成onnx的 我转的时候提示maxunpool不支持

Fight-hawk avatar Sep 15 '21 12:09 Fight-hawk

忘了,但印象中是支持的,没有修改过代码。

cmdbug avatar Sep 15 '21 23:09 cmdbug

忘了,但印象中是支持的,没有修改过代码。

能问下你用的是哪个版本的torch么

Fight-hawk avatar Sep 16 '21 05:09 Fight-hawk

挺久以前的,具体忘了。

cmdbug avatar Sep 16 '21 09:09 cmdbug

挺久以前的,具体忘了。

-.- 能回忆一下么 研究好几天了 始终没找到解决方案 非常感谢

Fight-hawk avatar Sep 16 '21 09:09 Fight-hawk

https://github.com/cmdbug/ENet 对比下这个工程,看下有没有区别。

cmdbug avatar Sep 16 '21 09:09 cmdbug

非常感谢 是去掉了maxunpool操作 建议更改一下enet的参考工程哦

Fight-hawk avatar Sep 16 '21 09:09 Fight-hawk

突然想起来可以试试 nihui 的 github 中的 ncnn 里面的 pnnx 分支,直接导出 torchscript -> pnnx -> ncnn。应该不会出现不支持的问题。

cmdbug avatar Oct 12 '21 12:10 cmdbug