mirror of
https://gitee.com/bianbu-linux/ai-support
synced 2025-04-19 11:54:49 -04:00
36 lines
No EOL
839 B
C++
36 lines
No EOL
839 B
C++
#ifndef SUPPORT_PYTHON_UTILS_H_
|
|
#define SUPPORT_PYTHON_UTILS_H_
|
|
|
|
#include <pybind11/numpy.h>
|
|
#include <pybind11/pybind11.h>
|
|
|
|
#include <string>
|
|
|
|
#include "opencv2/opencv.hpp"
|
|
namespace py = pybind11;
|
|
struct Box {
|
|
int x1;
|
|
int y1;
|
|
int x2;
|
|
int y2;
|
|
float score;
|
|
std::string label_text;
|
|
unsigned int label;
|
|
Box(int x1 = 0, int y1 = 0, int x2 = 0, int y2 = 0, float score = 0.0,
|
|
std::string label_text = "", unsigned int label = 0)
|
|
: x1(x1),
|
|
y1(y1),
|
|
x2(x2),
|
|
y2(y2),
|
|
score(score),
|
|
label_text(label_text),
|
|
label(label) {}
|
|
};
|
|
|
|
cv::Mat numpy_uint8_3c_to_cv_mat(const py::array_t<unsigned char> &input) {
|
|
py::buffer_info buf = input.request();
|
|
cv::Mat mat(buf.shape[0], buf.shape[1], CV_8UC3, (unsigned char *)buf.ptr);
|
|
return mat;
|
|
}
|
|
|
|
#endif // SUPPORT_PYTHON_UTILS_H_
|