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

    如何导出3588s模型

    Scheduled Pinned Locked Moved AI Algorithm
    8 Posts 2 Posters 552 Views
    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

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

      zhengbichengZ 1 Reply Last reply Reply Quote 1
      • zhengbichengZ
        zhengbicheng @ccjjww1222
        last edited by

        @ccjjww1222

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

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

        C 1 Reply Last reply Reply Quote 1
        • C
          ccjjww1222 @zhengbicheng
          last edited by

          @zhengbicheng

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

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

          zhengbichengZ 2 Replies Last reply Reply Quote 0
          • zhengbichengZ
            zhengbicheng @ccjjww1222
            last edited by

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

            1 Reply Last reply Reply Quote 0
            • zhengbichengZ
              zhengbicheng @ccjjww1222
              last edited by

              @ccjjww1222

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

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

              改一下以上两点

              C 1 Reply Last reply Reply Quote 0
              • C
                ccjjww1222 @zhengbicheng
                last edited by

                @zhengbicheng

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

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

                zhengbichengZ 1 Reply Last reply Reply Quote 0
                • zhengbichengZ
                  zhengbicheng @ccjjww1222
                  last edited by zhengbicheng

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

                  C 1 Reply Last reply Reply Quote 1
                  • C
                    ccjjww1222 @zhengbicheng
                    last edited by

                    @zhengbicheng 可以了 谢谢

                    1 Reply Last reply Reply Quote 0
                    • 1 / 1
                    • First post
                      Last post