Caffe
blankai
这个作者很懒,什么都没留下…
展开
-
在Caffe源码中增加自己定义的layer定义
(1)书写 include/caffe/layers/xxx_layer.hpp ; (2) 书写src/caffe/layers/xxx_layer.cpp ; (3) (optional)若需要在GPU上实现,书写src/caffe/layers/xxx_layer.cu ; (4) 修改src/caffe/layer_factory.cpp,增加这个函数 // Get pool...原创 2018-04-12 10:24:41 · 256 阅读 · 0 评论 -
caffe训练模型时segmentation fault(core dumped)停掉的解决方法
重新编译caffe的步骤: make clean; make all -j4 ; make test ; make runtest ; make pycaffe 报错:添加完新定义的layer文件后,在重新编译caffe后,在训练模型的时候直接segmentation fault(core dumped)停掉,也不说是哪错了 解决:重新编译caffe,要从make clean开始...原创 2018-04-25 11:38:38 · 3346 阅读 · 0 评论 -
proto/caffe.pb.h:17:2: error: #error This file was generated by an older version of protoc
重新编译caffe的步骤: make clean; make all -j4 ; make test ; make runtest ; make pycaffe 报错:重新编译protobuf新版本后,在编译caffe的时候报错proto/caffe.pb.h:17:2: error: #error This file was generated by an older v...原创 2018-04-25 11:40:46 · 1400 阅读 · 0 评论 -
caffe中添加新层的步骤
caffe中添加新层的步骤 1、增加caffe/include/caffe/layers/.hpp文件 2、增加caffe/src/caffe/layers/.cpp文件,若要用GPU,还要写.cu文件 3、修改src/caffe/proto/caffe.proto文件:改两个地方,如下 3.1一个是在message LayerParameter {}中添加一个唯一ID号,注意不要重...原创 2018-04-25 11:49:32 · 766 阅读 · 0 评论