读取caffemodel

最新推荐文章于 2021-05-06 18:19:32 发布

被自己蠢哭了

最新推荐文章于 2021-05-06 18:19:32 发布

阅读量694

点赞数

分类专栏：深度学习

本文链接：https://blog.youkuaiyun.com/jicong44/article/details/80771212

版权

深度学习专栏收录该内容

34 篇文章

订阅专栏

是在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