open-avm 的rknn推理

open-amv

open-amv视觉AI开发板是基于瑞芯微rv1103的主控芯片,外围设备丰富,上手简单,有0.5T的算力,轻松上手yolov5s,做目标检测非常适合。

rknn推理

rknn格式类似于pt,onnx,是由瑞芯微推出正对主控芯片的npu做推理的框架。我们需要把训练的模型用RKNN-Toolkit转为rknn格式,然后用rknn的提供API接口加载模型,设置模型参数,推理。
步骤

  1. 首先下载rknn转化工具: https://github.com/airockchip/rknn-toolkit2
    使用里面的docker安装很方便,选准平台。
  2. https://github.com/airockchip/yolov5 使用瑞芯微官网的yolov5训练模型,生成PT文件,python export.py --rknpu --weight yolov5s.pt导出onnx模型,然后用rknn-toolkit转化工具python3 convert.py …/model/yolov5s.onnx rv1106 转成rv1103 的rknn模型。
  3. 注意要rknn要量化8int,因为rv1103只支持量化的rknn模型
  4. 使用rknn的推理框架API调用进行推理

部署c++

从摄像头读取数据,图片处理转化模型需要的图片格式输入,推理获取结果做nms获取目标像素坐标,画框。
采用32bit的opencv读取摄像头和图片处理格式nchw,用mpp加速做图片预处理等比缩放,扔进去模型做推理,获取结果后用opencv画框。
具体参考链接: https://pan.baidu.com/s/12UqXwZjxkcyuPuulcjLHig?pwd=rcay 提取码: rcay 里面的npu测试例子。

  • 6
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值