首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >caffe layer zero输出-python

caffe layer zero输出-python
EN

Stack Overflow用户
提问于 2016-12-27 16:23:29
回答 0查看 243关注 0票数 1

我得到了VGG改变的网络的输出(这里是3个浮点数的向量)一切都好,但是当我只想得到其他层的输出(例如最后一个卷积层conv5_4)而不是最后一层时,输出向量都是零。以下是用于获取输出的python代码:

代码语言:javascript
复制
import sys
import cv2
import numpy as np
sys.path.insert(0, '/home/deep/rahim/caffe-master/python')
import caffe
input_image_file = sys.argv[1]
output_file = sys.argv[2]
model_file = '/home/deep/rahim/PGM/Final/feature/deep/snapshot_iter_355000.caffemodel'
deploy_prototxt = '/home/deep/rahim/PGM/Final/feature/deep/deploy.prototxt'
net = caffe.Net(deploy_prototxt, model_file, caffe.TEST)
layer = 'conv5_4'
transformer = caffe.io.Transformer({'data':       net.blobs['data'].data.shape})
transformer.set_transpose('data', (2,0,1))
transformer.set_raw_scale('data', 255.0)
net.blobs['data'].reshape(1,3,40,40)
img = caffe.io.load_image(input_image_file)
net.blobs['data'].data[...] = transformer.preprocess('data', img)
output = net.forward()
print '%s' %(net.blobs[layer].data[0])

请帮帮我

EN

回答

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/41341529

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档