count_caffemodel_params.py

#!/usr/bin/env python

# usage: python count_caffemodel_params.py /path/to/my.caffemodel [/path/to/my/other.caffemodel, ...]

from caffe.proto import caffe_pb2
import sys

assert len(sys.argv) >= 2

for caffemodel_filename in sys.argv[1:]:
    with open(caffemodel_filename, 'r') as caffemodel_file:
        caffemodel = caffemodel_file.read()
    net = caffe_pb2.NetParameter()
    net.ParseFromString(caffemodel)
    param_count = 0
    for layer in net.layer:
        for blob in layer.blobs:
            param_count += len(blob.data)
    print '%s: %d parameters' % (caffemodel_filename, param_count)