rk3399板子的模型移植方法

首先,需要先搭建环境和依赖项。

pip3 install rknn_toolkit-1.7.0-cp37-cp37m-linux_aarch64.whl --user
rknn_toolkit-1.7.0-cp37-cp37m-linux_aarch64.whl该驱动一般自带。

如果pip3没安装的话,还需要安装。(不要用conda安装rknn)

接下来还需要安装torch等依赖,参考资料给出了如下所示的安装命令:

pip3 install torch==1.5.1 torchvision==0.4.0

推荐使用镜像命令安装,如下所示,给出了一个豆瓣的镜像:

-f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.douban.com/simple

然后,要准备好onnx的模型,该模型要求opset_version<=11,因为经测试,pt之类的无法直接转成rnkk。实测发现连11版本都不行,只能11以下。

找到rnkk安装包\rknn-toolkit-1.7.0\examples\onnx\yolov5

python3 test.py

 

posted @ 2022-08-04 11:31  澳大利亚树袋熊  阅读(267)  评论(0)    收藏  举报