check net-blob

# cat check_size.py
import numpy as np

import sys
caffe_root = './'  # this file should be run from {caffe_root}/examples (otherwise change this line)
sys.path.insert(0, caffe_root + 'python')

import caffe
caffe.set_device(0)  # if we have multiple GPUs, pick the first one
caffe.set_mode_gpu()

model_def = caffe_root + '20180905_1.prototxt'
net = caffe.Net(model_def,   caffe.TEST)

for layer_name, blob in net.blobs.iteritems():
    print layer_name + '\t' + str(blob.data.shape)



for layer_name, param in net.params.iteritems():
    print layer_name + '\t' + str(param[0].data.shape), str(param[1].data.shape)

  

posted on 2019-08-16 17:53  cdekelon  阅读(107)  评论(0)    收藏  举报

导航