FazBrowse GitHub Viewer | Trending |
URL:
| Home
Tools: [Download Repo ZIP]   [Original HTTPS Page]

Merge pull request #4347 from nitnelave/python/layer_dict · rootkit/CaffeOnACL@c510e33 · GitHub

Repository navigation

Commit c510e33

Browse files
authored
Merge pull request #4347 from nitnelave/python/layer_dict
[pycaffe] add layer_dict to the python interface
2 parents 28ffe9c + 5417f10 commit c510e33

2 files changed

Lines changed: 18 additions & 0 deletions

File tree

‎python/caffe/pycaffe.py‎

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -43,6 +43,16 @@ def _Net_blob_loss_weights(self):
4343
self._blob_loss_weights))
4444
return self._blob_loss_weights_dict
4545

46+
@property
47+
def _Net_layer_dict(self):
48+
"""
49+
An OrderedDict (bottom to top, i.e., input to output) of network
50+
layers indexed by name
51+
"""
52+
if not hasattr(self, '_layer_dict'):
53+
self._layer_dict = OrderedDict(zip(self._layer_names, self.layers))
54+
return self._layer_dict
55+
4656

4757
@property
4858
def _Net_params(self):
@@ -321,6 +331,7 @@ def get_id_name(self):
321331
# Attach methods to Net.
322332
Net.blobs = _Net_blobs
323333
Net.blob_loss_weights = _Net_blob_loss_weights
334+
Net.layer_dict = _Net_layer_dict
324335
Net.params = _Net_params
325336
Net.forward = _Net_forward
326337
Net.backward = _Net_backward

‎python/caffe/test/test_net.py‎

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -60,6 +60,13 @@ def test_memory(self):
6060
for bl in blobs:
6161
total += bl.data.sum() + bl.diff.sum()
6262

63+
def test_layer_dict(self):
64+
layer_dict = self.net.layer_dict
65+
self.assertEqual(list(layer_dict.keys()), list(self.net._layer_names))
66+
for i, name in enumerate(self.net._layer_names):
67+
self.assertEqual(layer_dict[name].type,
68+
self.net.layers[i].type)
69+
6370
def test_forward_backward(self):
6471
self.net.forward()
6572
self.net.backward()

0 commit comments

Comments
 (0)

Back | FazBrowse Home | New Git URL