• Recent
  • Wiki
  • Github
  • 百度网盘
  • Onedrive
  • Official
  • Shop
  • Register
  • Login
  • Register
  • Login
  • Search
  • Recent
  • Wiki
  • Github
  • 百度网盘
  • Onedrive
  • Official
  • Shop

如何导出3588s模型

AI Algorithm
2
8
631
Loading More Posts
  • Oldest to Newest
  • Newest to Oldest
  • Most Votes
Reply
  • Reply as topic
Log in to reply
This topic has been deleted. Only users with topic management privileges can see it.
  • C
    ccjjww1222 last edited by Oct 23, 2022, 1:45 AM

    直接通过rknn-toolkit2导出的yolov5的rknn模型,在3588s上使用RKNNLite无法init_runtime。
    代码如下:
    33ae75e9-f63e-4c9b-94d2-2320558c195d-image.png
    报错如下:
    657cde07-8ba2-4d7f-871e-e47f38187360-image.png

    zhengbicheng 1 Reply Last reply Oct 23, 2022, 1:55 AM Reply Quote 1
    • zhengbicheng
      zhengbicheng @ccjjww1222 last edited by Oct 23, 2022, 1:55 AM

      @ccjjww1222

      • NPU需要系统sudo权限,请使用sudo -E python3 xxx.py来运行代码。

      • 使用yolov5请注意版本,目前测试yolov5 6.0的代码存在bug。可以考虑使用RK官方改进的yolov5做代替。

      C 1 Reply Last reply Oct 23, 2022, 2:05 AM Reply Quote 1
      • C
        ccjjww1222 @zhengbicheng last edited by Oct 23, 2022, 2:05 AM

        @zhengbicheng

        • 是在运行的时候sudo? 我在3588s板子上面运行已经报同样的错误。
          4b2aa4f9-0205-428d-b75e-c4545ec5659b-image.png

        • 模型我们会自己重新训练,目前使用yolov5仅仅是用来测试功能以及性能是否能满足

        zhengbicheng 2 Replies Last reply Oct 23, 2022, 2:33 AM Reply Quote 0
        • zhengbicheng
          zhengbicheng @ccjjww1222 last edited by Oct 23, 2022, 2:25 AM

          @ccjjww1222 你驱动安装了吗?参考我之前给你发的教程,里面有驱动的安装教程

          1 Reply Last reply Reply Quote 0
          • zhengbicheng
            zhengbicheng @ccjjww1222 last edited by Oct 23, 2022, 2:33 AM

            @ccjjww1222

            • 看你的log,你模型导出的device是RK356X系列的,需要改为RK3588。

            • RKNPU2目前只支持输入数据为NHWC格式

            改一下以上两点

            C 1 Reply Last reply Oct 23, 2022, 6:34 AM Reply Quote 0
            • C
              ccjjww1222 @zhengbicheng last edited by Oct 23, 2022, 6:34 AM

              @zhengbicheng

              • 驱动我装了,从这拉的代码https://github.com/rockchip-linux/rknpu2
                那这个路径下的文件使用的:/home/coolpi/rknpu2/runtime/RK3588/Linux/librknn_api/aarch64

              • RKNPU2目前只支持输入数据为NHWC格式----这个问题怎么改?

              zhengbicheng 1 Reply Last reply Oct 23, 2022, 6:44 AM Reply Quote 0
              • zhengbicheng
                zhengbicheng @ccjjww1222 last edited by zhengbicheng Oct 23, 2022, 7:05 AM Oct 23, 2022, 6:44 AM

                @ccjjww1222 NHWC这个配置不需要更改,输入图片不做nhwc转nchw的preprocess就行了。重要的是你需要更改目标板子的型号,你导出的模型是RK356X的,所以在RK3588上跑不通。请查询一下rknn-toolkit的文档,没记错的话应该是config函数存在这个配置选项。

                C 1 Reply Last reply Oct 23, 2022, 8:38 AM Reply Quote 1
                • C
                  ccjjww1222 @zhengbicheng last edited by Oct 23, 2022, 8:38 AM

                  @zhengbicheng 可以了 谢谢

                  1 Reply Last reply Reply Quote 0
                  • 1 / 1
                  5 out of 8
                  • First post
                    5/8
                    Last post