读取caffemodel

是在tools里面运行

import _init_paths
import numpy as np
import caffe

prototxt='/home/jc/py-R-FCN/models/pascal_voc/ResNet-101/rfcn_end2end/test_agnostic.prototxt'
caffemodel='/home/jc/py-R-FCN/output/rfcn_end2end_ohem/voc_2007_trainval/resnet101_rfcn_ohem_iter_90000.caffemodel'
net = caffe.Net(prototxt, caffemodel, caffe.TEST)

print net.blobs

import os
import _init_paths
import numpy as np
import caffe

prototxt='/home/jc/py-R-FCN/models/pascal_voc/ResNet-101/rfcn_end2end/test_agnostic.prototxt'
caffemodel='/home/jc/py-R-FCN/output/rfcn_end2end_ohem/voc_2007_trainval/resnet101_rfcn_ohem_iter_90000.caffemodel'
net = caffe.Net(prototxt, caffemodel, caffe.TEST)

blob=open('/home/jc/py-R-FCN/tools/blobs.txt','r')
output=open('/home/jc/py-R-FCN/tools/blobs_lines.txt','w')
lines=blob.readlines()
for i,line in enumerate(lines):
    for j , num in enumerate(line.split('), (')[:]):
        print num.split(',')[0][1:-1]
        output.writelines(num.split(',')[0]+'..............'+str(net.blobs[str(num.split(',')[0][1:-1])].data.shape)+'\n')
        print net.blobs[str(num.split(',')[0][1:-1])].data.shape