Toybrick

关于rknn_api C++版的模型输出问题?

zengwubin

中级会员

积分
274
楼主
发表于 2020-5-7 17:51:58    查看: 5631|回复: 1 | [复制链接]    打印 | 只看该作者
MTCNN的generateBbox函数,模型输出的score,loc有c h w属性:
vector<FaceInfo> MtcnnDetector::generateBbox(ncnn::Mat score, ncnn::Mat loc, float scale, float thresh)
{
    int stride = 2;
    int cellsize = 12;//人脸的最小检测范围
    float *p = score.channel(1);//得到是否是人脸的概率
    float inv_scale = 1.0f / scale;//放大的倍率
    vector<FaceInfo> results;//用来存放生成候选框的集合
    // printf("sssssssssss %d, %d \n", score.h, score.w);
    for (int row = 0; row < score.h; row++)
    {
        for (int col = 0; col < score.w; col++)
        {
            if (*p > thresh)//人脸的概率大于阈值才生产候选框

在NPU输出的是一维数组,还是python版本的4倍:
        float* location =(float*)outputs_pnet[0].buf;
        float* score = (float*)outputs_pnet[1].buf;
后处理该怎么做,怎么reshape成CPU推理时相应的维度呢??

回复

使用道具 举报

zengwubin

中级会员

积分
274
沙发
 楼主| 发表于 2020-5-7 18:33:50 | 只看该作者
对rknn_api输出的数据格式该怎么处理, 有没相关文档说明的
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

产品中心 购买渠道 开源社区 Wiki教程 资料下载 关于Toybrick


快速回复 返回顶部 返回列表