RKNNToolkit2 推理数据输入问题

RKNNToolkit2 推理数据输入问题

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

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

在这里插入图片描述

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

在这里插入图片描述

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

在这里插入图片描述

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

在这里插入图片描述

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

注意保存原图片!!!!

注意保存原图片!!!!

注意保存原图片!!!!

重要的事情说三遍 !

相关推荐

  1. 数据输入输出(C++)

    2024-06-11 13:08:04       59 阅读

最近更新

  1. docker php8.1+nginx base 镜像 dockerfile 配置

    2024-06-11 13:08:04       98 阅读
  2. Could not load dynamic library ‘cudart64_100.dll‘

    2024-06-11 13:08:04       106 阅读
  3. 在Django里面运行非项目文件

    2024-06-11 13:08:04       87 阅读
  4. Python语言-面向对象

    2024-06-11 13:08:04       96 阅读

热门阅读

  1. 廉价耐储存食物推荐: 末日生存爱好者

    2024-06-11 13:08:04       33 阅读
  2. C++day5

    C++day5

    2024-06-11 13:08:04      29 阅读
  3. 15年老程序员的内心独白

    2024-06-11 13:08:04       26 阅读
  4. 前端学习笔记(一)

    2024-06-11 13:08:04       27 阅读
  5. c语言基础篇C

    2024-06-11 13:08:04       32 阅读
  6. SpringMVC

    SpringMVC

    2024-06-11 13:08:04      30 阅读
  7. 小抄 20240608

    2024-06-11 13:08:04       24 阅读
  8. Debian 12.5 一键安装 Oracle 19C 单机

    2024-06-11 13:08:04       30 阅读
  9. iOS 查看runtime源码的几种方法

    2024-06-11 13:08:04       34 阅读