layer_features.py 文件源码

python
阅读 37 收藏 0 点赞 0 评论 0

项目:fast-image-retrieval 作者: xueeinstein 项目源码 文件源码
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)
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号