在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/路径下
重新编译即可
网友评论