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