如何导出3588s模型
-
直接通过rknn-toolkit2导出的yolov5的rknn模型,在3588s上使用RKNNLite无法init_runtime。
代码如下:
报错如下:
-
-
NPU需要系统sudo权限,请使用sudo -E python3 xxx.py来运行代码。
-
使用yolov5请注意版本,目前测试yolov5 6.0的代码存在bug。可以考虑使用RK官方改进的yolov5做代替。
-
-
-
是在运行的时候sudo? 我在3588s板子上面运行已经报同样的错误。
-
模型我们会自己重新训练,目前使用yolov5仅仅是用来测试功能以及性能是否能满足
-
-
@ccjjww1222 你驱动安装了吗?参考我之前给你发的教程,里面有驱动的安装教程
-
-
-
驱动我装了,从这拉的代码https://github.com/rockchip-linux/rknpu2
那这个路径下的文件使用的:/home/coolpi/rknpu2/runtime/RK3588/Linux/librknn_api/aarch64 -
RKNPU2目前只支持输入数据为NHWC格式----这个问题怎么改?
-
-
@ccjjww1222 NHWC这个配置不需要更改,输入图片不做nhwc转nchw的preprocess就行了。重要的是你需要更改目标板子的型号,你导出的模型是RK356X的,所以在RK3588上跑不通。请查询一下rknn-toolkit的文档,没记错的话应该是config函数存在这个配置选项。
-
@zhengbicheng 可以了 谢谢