caffe导出可视化中间层特征

最新推荐文章于 2020-12-03 12:00:56 发布

wait a minutes

最新推荐文章于 2020-12-03 12:00:56 发布

阅读量289

点赞数

分类专栏： # Caffe 深度学习框架文章标签：可视化 caffe

本文链接：https://blog.csdn.net/qq_35975447/article/details/108707436

版权

深度学习框架同时被 2 个专栏收录

25 篇文章 0 订阅

订阅专栏

Caffe

6 篇文章 0 订阅

订阅专栏

0.环境

ubuntu16.04
caffe

1.代码

# cpu caffe
caffe_root = '/caffe/'
sys.path.insert(0, caffe_root + 'python')  
import caffe

# gpu caffe
caffe_root = '/caffe_gpu/'
sys.path.insert(0, caffe_root + 'python')  
import caffe
caffe.set_device(0)

# init
net = caffe.Net(model_prototxt, model_weights, caffe.TEST)
net.blobs['blob1'].reshape(1, 3, height, width)

# forward get feature
net.blobs['input'].data[...] = image_input
reid_net.forward()['output']
featmap = self.reid_net.blobs['mid'].data[...]

其中mid可以为中间层的任意层。后面可视化部分参考：https://github.com/KaiyangZhou/deep-person-reid/blob/master/tools/visualize_actmap.py