RK3399 Pro C++加载两个不同模型

#include "rknn_api.h"
#include <thread>
#include <string>
#include <iostream>
#include <fstream>
#include "opencv2/core/core.hpp"
#include "opencv2/imgproc/imgproc.hpp"
#include "opencv2/highgui/highgui.hpp"

using namespace std;

void YOLO(string model_path)
{
   
    FILE *fp_ = fopen(model_path.c_str(), "rb");
    if (fp_ == NULL)
    {
   
        printf("fopen %s fail!\n", model_path.c_str());
        return;
    }
    fseek(fp_, 0, SEEK_END);
    int model_len = ftell(fp_);
    void *model_ = malloc(model_len);
    fseek(fp_, 0, SEEK_SET);
    if (model_len != fread(model_, 1, model_len, fp_))
    {
   
        printf("fread %s fail!\n", model_path.c_str());
        free(model_);
        return;
    }
    rknn_context ctx = 0;
    int ret = rknn_init(&ctx, model_, model_len, RKNN_FLAG_PRIOR_MEDIUM);
    if (ret < 0)
    {
   
        printf("rknn_init fail! ret=%d\n", ret);
        if (ctx)
            rknn_destroy(ctx);
        if (model_)
            free(model_);
        if (fp_)
            fclose(fp_);
        return;
    }
    
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

刀么克瑟拉莫

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值