# 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)