def print_network_sizes(self, model_file):
net = caffe.Net(model_file, caffe.TRAIN)
for k, v in net.blobs.items():
print k, v.data.shape
微信
公众号
扫码关注公众号