ubuntu 16.04在caffe中添加新的层

2019-05-16  本文已影响0人  ccccxxxxcccc

在caffe中添加upsample层

下载upsample_layer的文件夹

将upsample_layer.hpp放在include/caffe/layers/路径下;

将upsample_layer.cpp和upsample_layer.cu放在src/caffe/layers/路径下

然后需要在src/caffe/proto/caffe.proto文件中添加关于UpsampleParameter的参数

然后重新编译之后就可以正常运行了。


在caffe中添加dw层

将conv_dw_layer.hpp放在include/caffe/layers/路径下

将conv_dw_layer.cpp和conv_dw_layer.cu放在src/caffe/layers/路径下

重新编译即可

上一篇下一篇

猜你喜欢

热点阅读