VehiclePlate.h
612 Bytes
#ifndef __AI_ENGINE_MODULE_VEHICLE_PLATE__
#define __AI_ENGINE_MODULE_VEHICLE_PLATE__
#include "acl/acl.h"
#include "acl/ops/acl_dvpp.h"
#include "../ai_platform/header.h"
#include "ai_engine_header.h"
#include <vector>
using namespace std;
class VehiclePlate
{
public:
VehiclePlate(/* args */);
~VehiclePlate();
int init(int dev_id, string model_dir);
vector<VehiclePlateResult> process(vector<DeviceMemory*> vec_gpuMem);
void release();
private:
int m_devId;
aclrtContext m_algorthim_ctx;
void* m_handle{nullptr};
};
#endif // __AI_ENGINE_MODULE_VEHICLE_PLATE__