From ab5e86e2babf22892a241ae06a93b3f389a23db1 Mon Sep 17 00:00:00 2001
From: Jonathan L Long <jonlong@cs.berkeley.edu>
Date: Tue, 30 Sep 2014 19:06:18 -0700
Subject: [PATCH] [pycaffe] expose Net::CopyTrainedLayersFrom as Net.copy_from

This allows finetuning from Python, for example.
---
 python/caffe/_caffe.cpp | 1 +
 python/caffe/_caffe.hpp | 3 +++
 2 files changed, 4 insertions(+)

diff --git a/python/caffe/_caffe.cpp b/python/caffe/_caffe.cpp
index 5a81a42..5696f26 100644
--- a/python/caffe/_caffe.cpp
+++ b/python/caffe/_caffe.cpp
@@ -147,6 +147,7 @@ BOOST_PYTHON_MODULE(_caffe) {
   bp::class_<PyNet, shared_ptr<PyNet> >(
       "Net", bp::init<string, string>())
       .def(bp::init<string>())
+      .def("copy_from",             &PyNet::CopyTrainedLayersFrom)
       .def("_forward",              &PyNet::Forward)
       .def("_backward",             &PyNet::Backward)
       .def("reshape",               &PyNet::Reshape)
diff --git a/python/caffe/_caffe.hpp b/python/caffe/_caffe.hpp
index 622e887..a15144b 100644
--- a/python/caffe/_caffe.hpp
+++ b/python/caffe/_caffe.hpp
@@ -96,6 +96,9 @@ class PyNet {
   inline void check_contiguous_array(PyArrayObject* arr, string name,
       int channels, int height, int width);
 
+  void CopyTrainedLayersFrom(const string filename) {
+    net_->CopyTrainedLayersFrom(filename);
+  }
   void Forward(int start, int end) { net_->ForwardFromTo(start, end); }
   void Backward(int start, int end) { net_->BackwardFromTo(start, end); }
   void Reshape() { net_->Reshape(); }
-- 
2.7.4