caffe新增層

轉載:https://blog.csdn.net/wanggao_1990/article/details/78863669 按照這個老哥的博客,實現了一遍,不過這個老哥實在window上實現的,我在ubntu實現的,基本成功了。 下邊記錄一下。 1.介紹 在使用Caffe時,可能已有的層不滿足需求,需要實現自己的層,最好的方式是修改caffe.proto文件,增加對應cpp、h、cu的聲明和實現
相關文章
相關標籤/搜索