#include "detector.h"
|
#include "ItsVehicleTypeRec.h"
|
|
#include <stdio.h>
|
#include <stdlib.h>
|
|
#include "csdk_struct.h"
|
|
namespace csdk_wrap
|
{
|
static void* pITSInstance;
|
|
int init_vehicle_its_detector(int sceneMode, int modelMode, char *keyPath, char *modelPath) {
|
// 初始化授权
|
int iInitFlag = ITS_ThreadInit(keyPath);
|
if(iInitFlag != OK)
|
{
|
return iInitFlag;
|
}
|
|
// 初始化配置
|
::ITS_Rec_Param itsParam;
|
itsParam.iSceneMode = sceneMode;
|
itsParam.iModelMode = modelMode;
|
|
// 初始化SDK
|
pITSInstance = ITS_VehicleRecInit(modelPath, iInitFlag, itsParam);
|
|
return pITSInstance == NULL ? 1 : 0;
|
}
|
|
cVehicleITSResult* vehicle_its_detect(int *plateIDCount, const cIMAGE *img, const cRECT *rect) {
|
if (!img) {
|
return NULL;
|
}
|
|
cVehicleITSResult *ppos = NULL;
|
|
::V_Image* pImage = new(V_Image);
|
pImage->pImageData = (BYTE*)(img->data);
|
pImage->iImageHeight = img->height;
|
pImage->iImageWidth = img->width;
|
pImage->eType = E_BGR;
|
|
::ITS_Vehicle_Result result;
|
memset(&result, 0, sizeof(ITS_Vehicle_Result));
|
|
// 设置识别区域
|
::S_Rect rcDetect = {rect->left, rect->top, rect->right, rect->bottom};
|
|
int iVehicleNum = ITS_VehicleTypeRec(pITSInstance, pImage, &rcDetect, &result);
|
|
if (iVehicleNum > 0){
|
ppos = (cVehicleITSResult*)malloc(iVehicleNum * sizeof(cVehicleITSResult));
|
*plateIDCount = iVehicleNum;
|
memcpy(ppos, result.tempResult, sizeof(ITS_Vehicle_Result_Single) * iVehicleNum);
|
}
|
|
return ppos;
|
}
|
|
void uninit_vehicle_its_detector() {
|
ITS_VehicleRecRelease(pITSInstance);
|
}
|
|
} // csdk_wrap
|
