[pycaffe] add converter for vectors of Nets

diff --git a/python/caffe/_caffe.cpp b/python/caffe/_caffe.cpp
index 5a81a42..ce7b726 100644 (file)
--- a/python/caffe/_caffe.cpp
+++ b/python/caffe/_caffe.cpp
@@ -189,6 +189,9 @@ BOOST_PYTHON_MODULE(_caffe) {
       .def("solve",        &PySGDSolver::Solve)
       .def("solve",        &PySGDSolver::SolveResume);
 
+  bp::class_<vector<shared_ptr<PyNet> > >("NetVec")
+      .def(bp::vector_indexing_suite<vector<shared_ptr<PyNet> >, true>());
+
   bp::class_<vector<PyBlob<float> > >("BlobVec")
       .def(bp::vector_indexing_suite<vector<PyBlob<float> >, true>());
 

diff --git a/python/caffe/_caffe.hpp b/python/caffe/_caffe.hpp
index 622e887..0ed0974 100644 (file)
--- a/python/caffe/_caffe.hpp
+++ b/python/caffe/_caffe.hpp
@@ -152,6 +152,11 @@ class PyNet {
   bp::dict raw_scale_;
   bp::dict channel_swap_;
 
+  // this is here only to satisfy boost's vector_indexing_suite
+  bool operator == (const PyNet &other) {
+      return this->net_ == other.net_;
+  }
+
  protected:
   // The pointer to the internal caffe::Net instance.
   shared_ptr<Net<float> > net_;