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

    获取推理结果坐标

    Scheduled Pinned Locked Moved AI Algorithm
    14 Posts 3 Posters 980 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.
    • zhengbichengZ
      zhengbicheng @ccjjww1222
      last edited by

      @ccjjww1222

      这个后处理部分,暂时我们还没有提供,网络上先找找,目前yolov5正在TODO计划中。

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

        @zhengbicheng 假如不考虑yolov5,我只需要推理结果的坐标。这个对我们很重要,后续功能的功能都是通过坐标来实现的

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

          @ccjjww1222 每一个模型的输出设计都是不一样的,后处理也不一样,没办法提供一个通用的后处理代码。

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

            @zhengbicheng 可能是我的描述不清楚,我们的模型依旧使用 yoloV5训练,用自己训练的模型,不是使用默认的yolov5 模型
            yolov5有大概的时间计划不? 1个月?1个季度?还是明年 ?

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

              @ccjjww1222 我懂你的意思,你们就是把yolov5 clone下来训练一遍嘛。如果你们没有改动模型结构的话,在rknn上推理是会出错的。请先用这个仓库的yolov5进行训练:EASY-EAI-Yolov5

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

                @ccjjww1222 yolov5计划在11月底前实现适配,下周可能会推出轻量级目标检测框架Picodet的适配,您可以尝试一下。

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

                  @zhengbicheng 好的
                  我参考了https://github.com/soloist-v/yolov5_for_rknn/blob/master/rknn_detect_yolov5.py里面的方案,画出来的矩形框位置不对,还是等等你们的吧。
                  模型训练目前先不急,我先把矩形框位置画对了,再训练

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

                    @ccjjww1222 这个yolov5应该是没问题的呀,你训练的yolov5是哪个版本的yolov5?

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

                      @zhengbicheng
                      通过https://github.com/rockchip-linux/rknn-toolkit/tree/master/examples/onnx/yolov5/yolov5s.onnx转换的。
                      另外一个事情,我用来测试的图片分辨率是810 X 1080,它有问题,
                      但是我接入摄像头,就没啥问题,摄像头每一帧的图片是480 X 640。
                      代码这边:https://github.com/jianwei/machine99/blob/main/detect/RKNN/detect.py (模型我没有上传)

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

                        @zhengbicheng 这个是我摄像头推理的时间:
                        2f323f64-799c-446a-9fb6-43d716d1bf37-image.png

                        inference time:当前这一帧的推理时间
                        avg_inference_time:以上所有帧平均推理时间
                        width:宽
                        height:高,
                        fps:fps
                        这样的推理时间正常不 ?
                        .

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

                          @ccjjww1222 正常,确实就是100ms左右

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

                            @ccjjww1222

                            有问题无非两种情况。

                            • 第一种情况是模型本身就有问题。
                            • 第二种情况是模型没问题,rknn推理有问题。

                            排查的方法有两种。

                            • 第一种:针对RKNN推理精度问题,rknn-toolkit提供了连板调试的方法,但是我们还未开放这个方法。
                            • 第二种:请先用onnxruntime跑一遍onnx模型的推理结果,如果onnx推理出来的结果出错,说明是模型本身的问题。如果onnx推理没错,rknn错了,你再留言。

                            针对这个仓库的话,RKNN精度应该是没问题的,这个RK那边已经测试过了。只可能是模型不匹配,毕竟不可能100%都检测出来。

                            花 1 Reply Last reply Reply Quote 1
                            • 花
                              花月 @zhengbicheng
                              last edited by

                              @zhengbicheng said in 获取推理结果坐标:

                              第一种:针对RKNN推理精度问题,rknn-toolkit提供了连板调试的方法,但是我们还未开放这个方法。

                              这个功能会开放吗,是硬件的更改吗?

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