RKNNToolkit2 推理数据输入问题
以前在玩RKNN的时候,下载对应的模型时,由于众多大佬帮忙开源了自己准好的模型和写好的各种后处理,所以当时只要脚本能运行,就不去深究代码的内容和模型。由于最近任务的调整,也慢慢开始关注后处理的撰写和模型的网络结构。
今天在执行yolov5face的rk3588推理的时候,出现了corddump,csdn后发现并不是我这个问题的方法。
因为我打开verbose,发现模型是正常推理的,并产生了结果
所以我断定是后处理出现了问题,我先观察模型的输入和输出
此时我注意到模型的输入是四维度的,而我在推理前,把维度图像增加了一个维度
在推理得到信息并经过后处理后,我需要将信息会知道图片,而此时图片已经不是三维度的了,进行cvtColor当然会报错,注释到这句话,对img在增加维度前进行copy操作,绘制的时候直接对img的copy文件绘制,最后成功运行。
注意保存原图片!!!!
注意保存原图片!!!!
注意保存原图片!!!!
重要的事情说三遍 !