RK3588安装rknn-toolkit-lite2

DLst_liu

已于 2024-08-12 20:31:19 修改

阅读量1.6k

点赞数 11

分类专栏： RK3588系列文章标签： linux python 嵌入式硬件 YOLO

于 2024-08-12 18:30:20 首次发布

本文链接：https://blog.csdn.net/qq_45022754/article/details/141138009

版权

2 篇文章

订阅专栏

一、下载rknn-toolkit2包：

sudo git clone https://gitclone.com/github.com/airockchip/rknn-toolkit2

进入到所在包的位置，然后解压：

unzip rknn-toolkit2-master.zip

进入rknn-toolkit-lite2所在路径：

cd rknn-toolkit2-master/rknn-toolkit-lite2/packages/

安装对应的版本：

pip install rknn_toolkit_lite2-2.0.0b0-cp38-cp38-linux_aarch64.whl

注：这里的架构是aarch64，所以安装rknn-toolkit-lite2，如果是其他的架构选择安装rknn-toolkit2

cd rknn-toolkit2-master/rknn-toolkit-lite2/examples/resnet18/

python test.py

报错，安装cv2

pip install opencv-python -i https://mirrors.aliyun.com/pypi/simple/
python test.py

再次报错：/usr/lib/librknnrt.so: undefined symbol: rknn_set_core_mask

此时更新rknn_server和librknnrt.so

cd rknn-toolkit2-master/rknpu2/runtime/Linux/
sudo cp librknn_api/aarch64/librknnrt.so /usr/lib
sudo cp -r rknn_server/aarch64/usr/bin/. /usr/bin

运行test.py

python test.py

成功！！！

由于上面这个包没有rknn-toolkit-lite2的yolov5测试代码。

下载鲁班猫RK系列板的AI应用代码：

git clone https://gitee.com/LubanCat/lubancat_ai_manual_code

进入yolov5路径：

cd dev_env/rknn_toolkit_lite2/examples/yolov5_inference/

运行：

python test.py

报错：维度的问题

解决方法：打开test.py，添加以下两句代码

再次运行，成功！！！