#ifndef _ITS_VEHICLE_TYPE_REC_H_
|
#define _ITS_VEHICLE_TYPE_REC_H_
|
#include "PlateType.h"
|
|
#define MAX_VEHICLE_NUM (5)
|
|
#define MAX_FACE_NUM (10)
|
|
#define MAX_CHAR_NUM (10)
|
|
#define OK (0) //初始化正确
|
#define DATA_OUT (-1) //SDK过期
|
#define KEY_ERROR (-2) //未找到加密狗
|
#define MODE_PATH_ERROR (-3) //mode文件错误
|
#define MAC_KEY_ERROR (-4) //mac地址加密文件错误
|
#define THREAD_ERROR (-5) //超过线程授权数目
|
|
|
#define VTR_API extern "C" __attribute__((visibility("default")))
|
|
/***************车身颜色******************/
|
typedef enum
|
{
|
E_VEHICLE_COLOR_NOT,//未知
|
E_VEHICLE_RED, //红色
|
E_VEHICLE_GREEN, //绿色
|
E_VEHICLE_BLUE, //蓝色
|
E_VEHICLE_PINK, //粉色
|
E_VEHICLE_BROWN, //棕色
|
E_VEHICLE_YELLOW, //黄色
|
E_VEHICLE_WHITE, //白色
|
E_VEHICLE_BLACK, //黑色
|
E_VEHICLE_GRAY, //灰色(银色)
|
E_VEHICLE_CYAN, //青色
|
}E_Vehicle_Color;
|
|
typedef enum
|
{
|
E_VEHICLE_DARK, //深色
|
E_VEHICLE_LIGHT, //浅色
|
}E_Vehicle_Bright;
|
|
typedef enum
|
{
|
E_RGB,
|
E_BGR,
|
E_YUV420,
|
}V_ImageType; //图片格式
|
|
typedef struct
|
{
|
int iImageHeight;
|
int iImageWidth;
|
V_ImageType eType;
|
unsigned char *pImageData;
|
}V_Image; //图像格式
|
|
|
typedef struct
|
{
|
int iLeft;
|
int iTop;
|
int iRight;
|
int iBottom;
|
}S_Rect;
|
|
|
typedef struct
|
{
|
int iPlateFlag; //是否识别到车牌
|
PlateInfor plateInfor; //车牌信息
|
S_Rect plateRect; //车牌位置
|
S_Rect tempVehicleRect; //车辆位置
|
|
char tempVehicleType[1024]; //车辆类型--总名称 jeep-大切诺基-2004
|
char tempVehicleBrand[1024]; //车辆品牌 jeep
|
char tempVehicleSub[1024]; //车辆子型号 大切诺基
|
char tempVehicleType1[1024]; //车辆类型 轿车 货车 客车 面包车
|
|
float fVDConf; //车辆检测置信度
|
|
float fConfdence; //车辆类型置信度
|
int iVehicleSubModel; //车型名称索引
|
E_Vehicle_Bright eVehicleBright; //车辆深浅色
|
E_Vehicle_Color eVehicleColor1;//车身主颜色
|
E_Vehicle_Color eVehicleColor2;//车身辅颜色
|
|
unsigned char*pCaptureImage; //视频模式识别下抓拍的图片数据，必须调用时开辟，图片识别模式下不必开辟
|
//开辟大小为输入图片宽高乘积的3倍,且视频模式下每次只输出一个识别结果
|
//(注：此抓拍结果图的存储顺若输入图像是YUV格式，则为RGBRGB，否则与输如图像顺序相同)
|
int iImageWidth; //视频模式下抓拍图像宽度
|
int iImageHeight; //视频模式下抓拍图像高度
|
|
}ITS_Vehicle_Result_Single;
|
|
|
typedef struct
|
{
|
ITS_Vehicle_Result_Single tempResult[MAX_VEHICLE_NUM];
|
int iNum;
|
float fLprTime;
|
float fVtrTime;
|
}ITS_Vehicle_Result;
|
|
|
typedef struct
|
{
|
int iSceneMode; //场景模式： 0---顶装 1----侧装 2---顶装路径 3---停车场 4----公安
|
int iModelMode; //读取模型方式：0---原始模型 1---修改模型
|
}ITS_Rec_Param;
|
|
|
|
//利用车牌识别，车辆检测进行车型识别
|
VTR_API int ITS_ThreadInit(char* fullKeyPath);
|
|
VTR_API int ITS_GetThreadNum(); //获得加密锁支持的线程数目
|
|
VTR_API void * ITS_VehicleRecInit(char* modePath, int& iInitFlag, ITS_Rec_Param its_param);//modePath为model文件夹的路径名
|
|
VTR_API int ITS_VehicleTypeRec_new_1(void* pInstance,V_Image* pVImage,S_Rect* roi,ITS_Vehicle_Result* pResult, char plateResult[64]);
|
|
VTR_API int ITS_VehicleTypeRec(void* pInstance,V_Image* pVImage,S_Rect* roi,ITS_Vehicle_Result* pResult);
|
|
VTR_API void ITS_VehicleRecRelease(void* pInstance);
|
|
VTR_API void ITS_BinPlate(V_Image* pSrcPlate, V_Image* pBinPlate, int plateType);
|
|
|
VTR_API void*FeatureExt_Init(char* modePath,int& iFeatureLen);
|
|
VTR_API void FeatureExt_Extract(void* pInstance, V_Image* pImage, S_Rect* vehicleRoi,float* fFeature);
|
|
VTR_API void FeatureExt_Release(void* pInstance);
|
|
|
|
|
#endif
|
