leon 5 months ago
parent
commit
d638d72241
1 changed files with 4 additions and 2 deletions
  1. 4 2
      src/interface.cpp

+ 4 - 2
src/interface.cpp

@@ -1,13 +1,15 @@
 #include <sstream>
 #include <iostream>
-#include <pybind11/stl.h>
 #include <pybind11/pybind11.h>
+#include <pybind11/stl.h>
+#include <pybind11/stl_bind.h>
 #include <pybind11/numpy.h>
 #include "opencv2/opencv.hpp"
 #include "infer.hpp"
 #include "resnet.hpp"
 #include "cpm.hpp"
 
+#define UNUSED(expr) do { (void)(expr); } while (0)
 
 using namespace std;
 
@@ -61,7 +63,7 @@ public:
 
     //! 2. cast cv::Mat to numpy.ndarray
     static handle cast(const cv::Mat& mat, return_value_policy, handle defval){
-        // UNUSED(defval);
+        UNUSED(defval);
 
         std::string format = format_descriptor<unsigned char>::format();
         size_t elemsize = sizeof(unsigned char);