caffe新增层

转载:https://blog.csdn.net/wanggao_1990/article/details/78863669 按照这个老哥的博客,实现了一遍,不过这个老哥实在window上实现的,我在ubntu实现的,基本成功了。 下边记录一下。 1.介绍 在使用Caffe时,可能已有的层不满足需求,需要实现自己的层,最好的方式是修改caffe.proto文件,增加对应cpp、h、cu的声明和实现
相关文章
相关标签/搜索