|
|
@@ -1,7 +1,8 @@
|
|
|
#include <sstream>
|
|
|
#include <iostream>
|
|
|
-#include "pybind11/pybind11.h"
|
|
|
-#include "pybind11/numpy.h"
|
|
|
+#include <pybind11/stl.h>
|
|
|
+#include <pybind11/pybind11.h>
|
|
|
+#include <pybind11/numpy.h>
|
|
|
#include "opencv2/opencv.hpp"
|
|
|
#include "infer.hpp"
|
|
|
#include "resnet.hpp"
|
|
|
@@ -60,7 +61,7 @@ public:
|
|
|
|
|
|
//! 2. cast cv::Mat to numpy.ndarray
|
|
|
static handle cast(const cv::Mat& mat, return_value_policy, handle defval){
|
|
|
- Py_UNUSED(defval);
|
|
|
+ UNUSED(defval);
|
|
|
|
|
|
std::string format = format_descriptor<unsigned char>::format();
|
|
|
size_t elemsize = sizeof(unsigned char);
|
