美文网首页
ubuntu 16.04在caffe中添加新的层

ubuntu 16.04在caffe中添加新的层

作者: ccccxxxxcccc | 来源:发表于2019-05-16 11:18 被阅读0次

在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/路径下

重新编译即可

相关文章

网友评论

      本文标题:ubuntu 16.04在caffe中添加新的层

      本文链接:https://www.haomeiwen.com/subject/dvbbaqtx.html