From 8db1400f0221d86976eaff2b70f6dbe892ce894f Mon Sep 17 00:00:00 2001 From: Jonathan L Long Date: Mon, 10 Mar 2014 21:51:25 -0700 Subject: [PATCH] Expose caffe.Net.params as an OrderedDict --- python/caffe/pycaffe.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/python/caffe/pycaffe.py b/python/caffe/pycaffe.py index 5d36489..8fbbcf9 100644 --- a/python/caffe/pycaffe.py +++ b/python/caffe/pycaffe.py @@ -15,6 +15,9 @@ class Net(CaffeNet): super(Net, self).__init__(param_file, pretrained_param_file) self._blobs = OrderedDict([(bl.name, bl) for bl in super(Net, self).blobs]) + self.params = OrderedDict([(lr.name, lr.blobs) + for lr in super(Net, self).layers + if len(lr.blobs) > 0]) @property def blobs(self): -- 2.7.4