Skip to content

Commit 8719549

Browse files
committed
Pycaffe interface update.
1 parent f23e802 commit 8719549

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

python/caffe/_caffe.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -325,7 +325,7 @@ BOOST_PYTHON_MODULE(_caffe) {
325325
bp::return_value_policy<bp::copy_const_reference>()))
326326
.add_property("_blobs", bp::make_function(&Net<Dtype>::blobs,
327327
bp::return_internal_reference<>()))
328-
.add_property("_layers", bp::make_function(&Net<Dtype>::layers,
328+
.add_property("layers", bp::make_function(&Net<Dtype>::layers,
329329
bp::return_internal_reference<>()))
330330
.add_property("_blob_names", bp::make_function(&Net<Dtype>::blob_names,
331331
bp::return_value_policy<bp::copy_const_reference>()))

python/caffe/pycaffe.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -52,12 +52,12 @@ def _Net_params(self):
5252
if len(lr.blobs) > 0])
5353

5454
@property
55-
def _Net_layers(self):
55+
def _Net_layers_dict(self):
5656
"""
5757
An OrderedDict (bottom to top, i.e., input to output) of network
5858
layers indexed by name
5959
"""
60-
return OrderedDict(zip(self._layer_names, self._layers))
60+
return OrderedDict(zip(self._layer_names, self.layers))
6161

6262

6363
@property
@@ -313,7 +313,7 @@ def __getitem__(self, name):
313313
return [id_to_name[i] for i in ids]
314314

315315
# Attach methods to Net.
316-
Net.layers = _Net_layers
316+
Net.layers_dict = _Net_layers_dict
317317
Net.blobs = _Net_blobs
318318
Net.blob_loss_weights = _Net_blob_loss_weights
319319
Net.params = _Net_params

0 commit comments

Comments
 (0)