src.cpp 1.05 KB
#include "./ai_engine_module/VehicleAnalysis.h"
#include "./ai_engine_module/include.h"
#include "./utils/logger.hpp"

int main() {

    VehicleAnalysis va;
    va.init(0, 16);

    string filename = "./test.jpg";

    aclrtStream stream = nullptr;
	ACL_CALL(aclrtCreateStream(&stream), ACL_SUCCESS, SY_FAILED);
    DvppProcess* dvpp = new DvppProcess();
	dvpp->InitResource(stream);

    const int batchsize = 1;

    vector<ImageData> vec_img;
    ImageData dvpp_data;
    for (int b = 0; b < batchsize; b++) {
        ImageData src;
        Utils::ReadImageFile(src, filename); //将二进制图像读入内存,并读取宽高信息

        ACL_CALL(dvpp->CvtJpegToYuv420sp(dvpp_data, src), SY_SUCCESS, SY_FAILED); //解码
        vec_img.push_back(dvpp_data);
    }

    printf("1 \n");

    va.detect(vec_img);


    delete dvpp;
    dvpp = nullptr;

    if (stream != nullptr) {
        int ret = aclrtDestroyStream(stream);
        if (ret != ACL_SUCCESS) {
            LOG_ERROR("destroy stream failed");
        }
        stream = nullptr;
    }

    return 0;
}