python调用c++代码例子,实现图像,class通信

例子运行环境:ubuntu14.04,cmake2.8,boost,opencv

c++ 代码:main.cpp

 1 #include 
 2 #include <string>
 3 #include 
 4 
 5 #include "boost/python.hpp"
 6 namespace bp = boost::python;
 7 
 8 typedef struct BBox {
 9     int x;
10     float confidence;
11 } _BBox;
12 
13 class communication{
14 public:
15     communication(bp::str str) {
16         imagename = std::string(((const char *) bp::extract<const char *>(str)));
17         std::cout << "communication init." << std::endl;
18         std::cout << "image name is " << imagename << std::endl;
19     }
20 
21     bp::list test(int rows,int cols,bp::str img_data) {
22         unsigned char *data = (unsigned char *) ((const char *) bp::extract<const char *>(img_data));
23         cv::Mat m= cv::Mat(rows, cols, CV_8UC3,data);
24         cv::cvtColor(m, m, cv::COLOR_BGR2GRAY);
25         cv::imshow(imagename, m);
26         cv::waitKey(1);
27 
28         bp::list img_list = bp::list();
29         for( size_t nrow = 0; nrow < m.rows; nrow++) {
30             unsigned char* data1 = m.ptrchar>(nrow);
31             for(size_t ncol = 0; ncol < m.cols; ncol++)
32                 img_list.appendchar>(data1[ncol]);
33         }
34         return img_list;
35     }
36     bp::list ret_box_list() {
37         bp::list ret_list = bp::list();
38         _BBox a,b;
39         a.x = 1;
40         a.confidence = 1.3;
41         b.x = 2;
42         b.confidence = 2.7;
43         ret_list.append(a);
44         ret_list.append(b);
45         return ret_list;
46     }
47 
48 
49     static void set_something(int id) {
50         std::cout << "set_something done." << std::endl;
51     }
52 
53     std::string imagename;
54 };
55 
56 BOOST_PYTHON_MODULE (interact) {
57     bp::class_("communication", bp::init())
58             .def("test", &communication::test)
59             .def("ret_box_list", &communication::ret_box_list)
60             .def("set_something", &communication::set_something)
61             .staticmethod("set_something");
62 
63     bp::class_("BBox")
64             .def_readonly("x", &BBox::x)
65             .def_readonly("confidence", &BBox::confidence);
66 }

cmake 代码:CMakeLists.txt

cmake_minimum_required(VERSION 2.8)
project(interact_cpp_py)

#aux_source_directory (${PROJECT_SOURCE_DIR} SRC_LIST)
#add_executable (${PROJECT_NAME} ${SRC_LIST})

## Compiler settings. generate .so needed by python
#set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -Wl,--export-dynamic -Wall -Wno-sign-compare -fPIC")

#BOOST
FIND_PACKAGE(Boost COMPONENTS python)
FIND_PACKAGE(PythonInterp)
FIND_PACKAGE(PythonLibs)
include_directories(SYSTEM ${Boost_INCLUDE_DIR})
include_directories(SYSTEM ${PYTHON_INCLUDE_DIR})
link_libraries(${BOOST_LIBRARIES} ${PYTHON_LIBRARIES})
#target_link_libraries(${PROJECT_NAME} ${LIBRARIES} ${Boost_LIBRARIES} ${PYTHON_LIBRARIES})

##OPENCV
find_package( OpenCV REQUIRED )
#target_link_libraries( ${PROJECT_NAME}
#    ${OpenCV_LIBS}
#)

## build .so
set(CMAKE_LIBRARY_OUTPUT_DIRECTORY ${PROJECT_BINARY_DIR}/lib)
add_library(interact SHARED main.cpp)
set_target_properties(interact PROPERTIES PREFIX "")
target_link_libraries(interact ${LIBRARIES} ${Boost_LIBRARIES} ${PYTHON_LIBRARIES} ${OpenCV_LIBS})

python代用c++代码:test.py

 1 import numpy as np
 2 import interact, cv2
 3 
 4 class call_cpp(object):
 5     def __init__(self, name):
 6         self.itcom = interact.communication(name)
 7 
 8     def run(self, img) :
 9         image_char = img.astype(np.uint8).tostring()
10         ret_list = self.itcom.test(img.shape[0], img.shape[1], image_char)
11         gray = np.zeros([img.shape[0], img.shape[1]], np.ubyte)
12         c = 0
13         for i in range(img.shape[0]):
14             for j in range(img.shape[1]):
15                 gray[i][j] = ret_list[c]
16                 c += 1
17         return gray
18 
19     def run_class(self):
20         return self.itcom.ret_box_list()
21 
22 if __name__ == '__main__':
23     image = cv2.imread('deng.jpg')
24     itr = call_cpp("itr")
25     a = itr.run_class()
26     print a[0].x, a[0].confidence, a[1].x, a[1].confidence
27     image1 = itr.run(image)
28     cv2.imshow("python", image1)
29     cv2.waitKey(0)

 

转载于:https://www.cnblogs.com/banrixianxin/p/6742364.html

你可能感兴趣的:(python调用c++代码例子,实现图像,class通信)