[pycaffe] expose Net.blob_names and Net.layer_names

diff --git a/python/caffe/_caffe.cpp b/python/caffe/_caffe.cpp
index 7bf5ab2..39e5d09 100644 (file)
--- a/python/caffe/_caffe.cpp
+++ b/python/caffe/_caffe.cpp
@@ -156,6 +156,8 @@ BOOST_PYTHON_MODULE(_caffe) {
       .def("set_device",            &PyNet::set_device)
       .add_property("_blobs",       &PyNet::blobs)
       .add_property("layers",       &PyNet::layers)
+      .add_property("_blob_names",  &PyNet::blob_names)
+      .add_property("_layer_names", &PyNet::layer_names)
       .add_property("inputs",       &PyNet::inputs)
       .add_property("outputs",      &PyNet::outputs)
       .add_property("mean",         &PyNet::mean_)

diff --git a/python/caffe/_caffe.hpp b/python/caffe/_caffe.hpp
index 2df47c7..5884900 100644 (file)
--- a/python/caffe/_caffe.hpp
+++ b/python/caffe/_caffe.hpp
@@ -123,6 +123,9 @@ class PyNet {
     return vector<PyLayer>(net_->layers().begin(), net_->layers().end());
   }
 
+  vector<string> blob_names() { return net_->blob_names(); }
+  vector<string> layer_names() { return net_->layer_names(); }
+
   bp::list inputs() {
     bp::list input_blob_names;
     for (int i = 0; i < net_->input_blob_indices().size(); ++i) {