def layer_features(layers, model_file, deploy_file, imagemean_file,
image_files, gpu=True, gpu_id=0, show_pred=False):
"""extract features from various layers"""
if gpu:
caffe.set_device(gpu_id)
caffe.set_mode_gpu()
net = feed_net(model_file, deploy_file, imagemean_file, image_files,
show_pred)
#if type(layers) == str:
#return net.blobs[layers].data
for layer in layers:
if layer not in net.blobs:
raise TypeError('Invalid layer name: ' + layer)
yield (layer, net.blobs[layer].data)
layer_features.py 文件源码
python
阅读 37
收藏 0
点赞 0
评论 0
评论列表
文章目录