msnh2012 / Msnhnet

🔥 (yolov3 yolov4 yolov5 unet ...)A mini pytorch inference framework which inspired from darknet.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

模型转换的疑问

huangzhenjie opened this issue · comments

commented

您好,这个项目非常好,我想请教下,不支持的op怎么添加吗?

你好,有哪些OP需要转换,模型最好是用pytorch原生的一些操作,尽量减少一些+-之列的拼凑

commented

就是比如一些后处理操作,比如维度的广播对齐,切片,升维降维,逻辑运算符之类的

后处理多么?不多的话可以自己实现下

commented

就是这些都是不支持转出来的吧

你可以对照下这页的OP: https://github.com/msnh2012/Msnhnet/tree/master/tools/pytorch2Msnhnet。
目前我在弄机器人相关的代码,所以暂时没有更新DL的代码了, 或者你可以试试NCNN

commented

好的,非常感谢