| | |
| | | #include "opencv2/imgproc/imgproc_c.h" // C |
| | | #endif // OPENCV |
| | | |
| | | //extern "C" { |
| | | //#include "image.h" |
| | | //} |
| | | |
| | | #ifdef YOLODLL_EXPORTS |
| | | #define YOLODLL_API __declspec(dllexport) |
| | | #else |
| | |
| | | #endif |
| | | |
| | | struct bbox_t { |
| | | float x, y, w, h; |
| | | unsigned int x, y, w, h; |
| | | float prob; |
| | | unsigned int obj_id; |
| | | }; |
| | |
| | | YOLODLL_API Detector(std::string cfg_filename, std::string weight_filename, int gpu_id = 0); |
| | | YOLODLL_API ~Detector(); |
| | | |
| | | YOLODLL_API std::vector<bbox_t> Detector::detect(std::string image_filename, float thresh = 0.2); |
| | | |
| | | YOLODLL_API std::vector<bbox_t> detect(std::string image_filename, float thresh = 0.2); |
| | | YOLODLL_API std::vector<bbox_t> detect(image_t img, float thresh = 0.2); |
| | | |
| | | |
| | | #ifdef OPENCV |
| | | std::vector<bbox_t> detect(cv::Mat mat, float thresh = 0.2) { |
| | | std::vector<bbox_t> detect(cv::Mat mat, float thresh = 0.2) |
| | | { |
| | | std::shared_ptr<image_t> image_ptr(new image_t, [](image_t *img) { free_image(*img); } ); |
| | | *image_ptr = mat_to_image(mat); |
| | | return detect(*image_ptr, thresh); |