python wrapper basic functionality ready

Author: for-aiur

.gitignore

@@ -11,3 +11,5 @@ bad.list
 Makefile
 CMakeCache.txt
 *.swp
+darknet
+darknet_cpp

app/darknet++/CMakeLists.txt

@@ -6,6 +6,8 @@ include_directories(${CUDA_INCLUDE_DIRS})
 
 find_package(PythonLibs 2.7 REQUIRED)
 find_package(Boost COMPONENTS python REQUIRED)
+include_directories(${PYTHON_INCLUDE_DIRS})
+include_directories(${Boost_INCLUDE_DIR})
 
 add_executable(darknet_cpp darknet++.cpp)
 target_link_libraries(

yolo++/CMakeLists.txt

@@ -25,9 +25,9 @@ find_package(PythonLibs 2.7 REQUIRED)
 include_directories(${PYTHON_INCLUDE_DIRS})
 find_package(Boost COMPONENTS python REQUIRED)
 include_directories(${Boost_INCLUDE_DIR})
-add_library(greet_ext SHARED greet_ext.cpp)
-target_link_libraries(greet_ext ${Boost_LIBRARIES} ${MODULENAME})
-set_target_properties(greet_ext PROPERTIES PREFIX "")
+add_library(yolopy SHARED yolopy.cpp)
+target_link_libraries(yolopy ${Boost_LIBRARIES} ${MODULENAME})
+set_target_properties(yolopy PROPERTIES PREFIX "")
 
 
 

yolo++/yolo.cpp

@@ -3,11 +3,41 @@
 
 #include <opencv2/highgui/highgui.hpp>
 
-std::string greet(float val, std::string str)
+YoloPython::YoloPython(){
+	yolo = new Yolo();
+   	yolo->setConfigFilePath("../../cfg/tiny-yolo-tayse.cfg");
+    	yolo->setDataFilePath("../../cfg/tayse.data");
+    	yolo->setWeightFilePath("/home/yildirim/Dropbox/tayse/models/tiny-yolo-tayse_16000.weights");
+    	yolo->setAlphabetPath("../../data/labels/");
+    	yolo->setNameListFile("../../data/tayse.names");
+    	yolo->setThreshold(0.16);
+}
+
+void YoloPython::setThreshold(float val){
+	yolo->setThreshold(val);
+}
+
+int YoloPython::detect(std::string path) {
+                cv::Mat img = cv::imread(path.c_str());
+		cv::resize(img, img, cv::Size(412,412));
+                yolo->detect(img, objects);
+                return objects.size();
+}
+
+float YoloPython::getComponent(int bb_idx, int component_idx) 
 {
-	Yolo yolo;
-	yolo.setThreshold(val);
-        return std::to_string(yolo.getThreshold());
+	if( bb_idx < 0 || bb_idx >= objects.size())
+		return 0.0;
+
+	if( component_idx == 0 )
+		return objects[bb_idx].bounding_box.x;
+	if( component_idx == 1 )
+		return objects[bb_idx].bounding_box.y;
+	if( component_idx == 2 )
+		return objects[bb_idx].bounding_box.width;
+	if( component_idx == 3 )
+		return objects[bb_idx].bounding_box.height;
+	return 0.0f;
 }
 
 Yolo::Yolo() : thresh(.24f),

yolo++/yolo.h

@@ -13,8 +13,9 @@
 #include <image.h>
 
 #include <object.h>
+#include <opencv2/highgui/highgui.hpp>
 
-std::string greet(float val, std::string path);
+#include <boost/python/numeric.hpp>
 
 class Yolo {
 public:
@@ -59,5 +60,14 @@ class Yolo {
     Yolo(const Yolo& copy);
 };
 
+struct YoloPython{
+	YoloPython();
+	void setThreshold(float val);
+	int detect(std::string path);
+	float getComponent(int bb_idx, int component_idx);
+	Yolo* yolo;
+        std::vector<DetectedObject> objects;
+};
+
 
 #endif //DARKNET_YOLO_H

yolo++/yolopy.cpp

@@ -0,0 +1,11 @@
+#include "yolo.h"
+#include <boost/python.hpp>
+
+BOOST_PYTHON_MODULE(yolopy)
+{
+	using namespace boost::python;
+	class_<YoloPython>("YoloPython")
+	.def("set_thresh", &YoloPython::setThreshold)
+	.def("comp", &YoloPython::getComponent)
+	.def("detect", &YoloPython::detect);
+}