RKNNToolkit2 推理数据输入问题

RKNNToolkit2 推理数据输入问题

​ 以前在玩RKNN的时候,下载对应的模型时,由于众多大佬帮忙开源了自己准好的模型和写好的各种后处理,所以当时只要脚本能运行,就不去深究代码的内容和模型。由于最近任务的调整,也慢慢开始关注后处理的撰写和模型的网络结构。

今天在执行yolov5face的rk3588推理的时候,出现了corddump,csdn后发现并不是我这个问题的方法。

在这里插入图片描述

因为我打开verbose,发现模型是正常推理的,并产生了结果

在这里插入图片描述

所以我断定是后处理出现了问题,我先观察模型的输入和输出

在这里插入图片描述

此时我注意到模型的输入是四维度的,而我在推理前,把维度图像增加了一个维度

在这里插入图片描述

在推理得到信息并经过后处理后,我需要将信息会知道图片,而此时图片已经不是三维度的了,进行cvtColor当然会报错,注释到这句话,对img在增加维度前进行copy操作,绘制的时候直接对img的copy文件绘制,最后成功运行。

注意保存原图片!!!!

注意保存原图片!!!!

注意保存原图片!!!!

重要的事情说三遍 !

  • 3
    点赞
  • 9
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值