diff options
-rw-r--r-- | Makefile | 4 | ||||
-rw-r--r-- | python/caffe/__init__.py | 2 | ||||
-rw-r--r-- | python/caffe/_caffe.cpp (renamed from python/caffe/pycaffe.cpp) | 2 |
3 files changed, 4 insertions, 4 deletions
@@ -45,8 +45,8 @@ NONGEN_CXX_SRCS := $(shell find \ LINT_REPORT := $(BUILD_DIR)/cpp_lint.log FAILED_LINT_REPORT := $(BUILD_DIR)/cpp_lint.error_log # PY$(PROJECT)_SRC is the python wrapper for $(PROJECT) -PY$(PROJECT)_SRC := python/$(PROJECT)/py$(PROJECT).cpp -PY$(PROJECT)_SO := python/$(PROJECT)/py$(PROJECT).so +PY$(PROJECT)_SRC := python/$(PROJECT)/_$(PROJECT).cpp +PY$(PROJECT)_SO := python/$(PROJECT)/_$(PROJECT).so # MAT$(PROJECT)_SRC is the matlab wrapper for $(PROJECT) MAT$(PROJECT)_SRC := matlab/$(PROJECT)/mat$(PROJECT).cpp MAT$(PROJECT)_SO := matlab/$(PROJECT)/$(PROJECT) diff --git a/python/caffe/__init__.py b/python/caffe/__init__.py index 9dd0ebf1..e8a8db7c 100644 --- a/python/caffe/__init__.py +++ b/python/caffe/__init__.py @@ -1 +1 @@ -from .pycaffe import * +from ._caffe import * diff --git a/python/caffe/pycaffe.cpp b/python/caffe/_caffe.cpp index ae0cdf06..1d7b2af3 100644 --- a/python/caffe/pycaffe.cpp +++ b/python/caffe/_caffe.cpp @@ -255,7 +255,7 @@ struct CaffeNet { // The boost python module definition. -BOOST_PYTHON_MODULE(pycaffe) { +BOOST_PYTHON_MODULE(_caffe) { boost::python::class_<CaffeNet>( "CaffeNet", boost::python::init<string, string>()) .def("Forward", &CaffeNet::Forward) |