#pragma once
|
|
#ifdef _WIN32
|
#ifdef AI_DETECT_EX_EXPORTS
|
#define AI_DETECT_EX_API __declspec(dllexport)
|
#else
|
#define AI_DETECT_EX_API __declspec(dllimport)
|
#endif
|
#else
|
#define AI_DETECT_EX_API
|
#endif
|
|
#include <string>
|
#include <vector>
|
|
class AI_DETECT_EX_API AiDetectResult
|
{
|
public:
|
int type;
|
char typeName[64];
|
|
double x;
|
double y;
|
double width;
|
double height;
|
|
double score;
|
}; // class AiDetectResult
|
|
namespace cv
|
{
|
class Mat;
|
} // namespace cv
|
|
class ProcessController;
|
class AI_DETECT_EX_API AiDetectEx
|
{
|
public:
|
static AiDetectEx& instance();
|
|
bool startServer();
|
void stopServer();
|
|
bool detect(cv::Mat const & srcImage, std::vector<AiDetectResult>& results, std::string channel, unsigned long timeOut = 0xFFFFFFFF);
|
bool detect(unsigned char* image, int width, int height, int step, std::vector<AiDetectResult>& results, std::string channel, unsigned long timeOut = 0xFFFFFFFF);
|
|
std::string generateChannel();
|
void releaseChannel(std::string channel);
|
|
AiDetectEx(const AiDetectEx&) = delete;
|
AiDetectEx& operator=(const AiDetectEx&) = delete;
|
private:
|
AiDetectEx();
|
~AiDetectEx();
|
|
bool isDisplay_;
|
ProcessController* controller_;
|
};
|
