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__