Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

ncnn2mem 如何加入自定义层? #304

Closed
jimmysue opened this issue Mar 24, 2018 · 6 comments
Closed

ncnn2mem 如何加入自定义层? #304

jimmysue opened this issue Mar 24, 2018 · 6 comments

Comments

@jimmysue
Copy link

我想通过 register_custom_layer 注册自定义层, 但是ncnn2mem 却只支持内部层, 请问这个时候这么做比较好?

@JACKYLUO1991
Copy link

楼上弄懂了吗?是否可以提示一下?

@jimmysue
Copy link
Author

最后我们写成了内部层应付一下

@ChenjingYu1993
Copy link

最后我们写成了内部层应付一下

写成内部层?能指导一下吗?我想把mobilessdlite模型打包一下,使用命令为:

ncnn2mem ssdlite.param ssdlite.bin ssd.id.h ssd.mem.h
生成ssd.id.h中出现两个Blob_data定义:
default
在网络载入时就报错了,说重定义,估计应该是模型的silence自定义层出现问题,请问有什么解决方案吗?

@nihui
Copy link
Member

nihui commented Sep 18, 2019

Silence conversion implemented in d67d253

For registering custom layer, you can refer https://github.com/Tencent/ncnn/blob/master/examples/mobilenetv2ssdlite.cpp

@nihui nihui closed this as completed Sep 18, 2019
@SophieChang66
Copy link

最后我们写成了内部层应付一下
你好,请问放入内部层,除了定义自定义层和增加ncnn_add_layer,还需要修改哪里?我加入内部后编译的库,普通的没有自定义的层都不能解析参数了

@SophieChang66
Copy link

最后我们写成了内部层应付一下
你好,请问放入内部层,除了定义自定义层和增加ncnn_add_layer,还需要修改哪里?我加入内部后编译的库,普通的没有自定义的层都不能解析参数了

自己回答一个,已解决,解决方式就是在layer下加入自定义层,然后在cmakelist.txt里加入自定义测ncnn_add_layer(),注意自定义定义层的头文件名必须和层名一致,大小写不同没有关系

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

5 participants