【yolov5系列】yolov5在瑞芯微RK3566上推理的所需链接记录

本来想写一下将yolov5移植到rk3566的全过程,结果发现github上面提供的文档比较仔细了。当前主板也不在手上,就只在这里做一下链接的记录。

  • 该文档链接为:https://github.com/rockchip-linux/rknn-toolkit2/blob/master/Rockchip_Quick_Start_RKNN_SDK_V1.4.0_CN.pdf

Rockchip NPU提供驱动、示例:

  • RK1808/RK1806/RV1109/RV1126,参考:https://github.com/rockchip-linux/rknpu
  • RK3566/RK3568/RK3588/RV1103/RV1106,参考:https://github.com/rockchip-linux/rknpu2
  • RK3399Pro用户态的库及驱动,参考:https://github.com/airockchip/RK3399Pro_npu

RKNN Toolkit:
在使用RKNN API进行部署之前,需要使用RKNN Toolkit将原始的模型转化成rknn模型。

  • RK1808/RK1806/RV1109/RV1126/RK3399Pro 使用: https://github.com/rockchip-linux/rknn-toolkit
  • RK3566/RK3568/RK3588/RV1103/RV1106使用:https://github.com/rockchip-linux/rknn-toolkit2

这里提下,NPU full driver与mini driver的区别(详细的驱动说明在 Rockchip NPU下的链接中可了解)。主要包含以下几点:

  • Mini driver只支持预编译的rknn模型,如果跑非预编译模型,会出现 RKNN_ERR_MODEL_INVALID的错误,从1.6.0开始,会返回RKNN_ERR_NON_PRE_COMPILED_MODEL_ON_MINI_DRIVER的错误;
  • Full driver支持RKNN Toolkit的联机调试功能,mini driver不支持;
  • Mini driver 库大小比full driver小很多,以RV1109/RV1126 1.6.0驱动为例,full driver大小为87MB,mini driver大小为7.1MB,可以有效的节省flash大小。
  • Mini driver库运行时占用的内存比full driver小。

你可能感兴趣的:(yolo系列,YOLO,rknn)