Toybrick

标题: yolov5的速度测试（更新原版yolov5结构导出） [打印本页]

作者: Zen 时间: 2021-3-29 16:15
标题: yolov5的速度测试（更新原版yolov5结构导出）
本帖最后由 Zen 于 2021-3-30 20:13 编辑

https://github.com/EASY-EAI/yolov5

我们这边测试的yolov5的运行时间，有需要的可以参考下哈，包含1808，1109，1126 三个平台。

======== 2021/3/30 更新 ========

链接中已经更新，可以导出原版 yolov5结构，包含 slice层、silu的处理。过几天还会更新速度测试。欢迎点赞留言提issue

作者: NoahCSHN 时间: 2021-4-8 08:51
你好，我想请教一下，你们代码只有将yolov5从torch转到onnx，没有从onnx转到rknn的过程是吗？

作者: sunkai 时间: 2021-4-9 09:06
您好，1126的板子，是在pc端连接板子，跑的程序吗？还是直接在板子上跑的

作者: Zen 时间: 2021-4-10 23:55

NoahCSHN 发表于 2021-4-8 08:51
你好，我想请教一下，你们代码只有将yolov5从torch转到onnx，没有从onnx转到rknn的过程是吗？ ...

是的，onnx到rknn没有特别多变化，后面有空会补一个。j

作者: Zen 时间: 2021-4-10 23:55

sunkai 发表于 2021-4-9 09:06
您好，1126的板子，是在pc端连接板子，跑的程序吗？还是直接在板子上跑的

直接在板子上跑哦

作者: sunkai 时间: 2021-4-12 08:48

Zen 发表于 2021-4-10 23:55
直接在板子上跑哦

rv1126上面可以直接跑python程序？

作者: sunkai 时间: 2021-4-12 09:00

sunkai 发表于 2021-4-12 08:48
rv1126上面可以直接跑python程序？

需要先进行编译吗？我之前测试的都是在pc上跑，程序中指定target = RV1126。直接在板子上跑能不能大概说下流程，感谢

作者: Zen 时间: 2021-4-12 14:27

sunkai 发表于 2021-4-12 09:00
需要先进行编译吗？我之前测试的都是在pc上跑，程序中指定target = RV1126。直接在板子上跑能不能大概说 ...

不是python，用 c 写的，基于 rknn_api 来调用 rknn模型

作者: sunkai 时间: 2021-4-12 14:55

Zen 发表于 2021-4-12 14:27
不是python，用 c 写的，基于 rknn_api 来调用 rknn模型

感谢，但是我在你发的仓库里面只看到了python版的，是没有贴出来的吗

作者: loveltyoic 时间: 2021-5-11 11:12

Zen 发表于 2021-4-12 14:27
不是python，用 c 写的，基于 rknn_api 来调用 rknn模型

请问又C++的示例代码吗？github上没看到呢

作者: chengzhang 时间: 2021-5-11 14:32
```
torch 1.7.1+cu101 RTX 2080 Ti
python 3.6
onnx 1.6
rknn-toolkit 1.6.0
```
训练的模型为yolov5s
运行命令：python models/export.py --rknn_mode --add_image_preprocess_layer
出错, 请问我应该怎么处理，如果使用 python models/export.py 是可以正常导出onnx模型的

```
Fusing layers...
Model Summary: 224 layers, 7053910 parameters, 0 gradients
Traceback (most recent call last):
  File "models/export.py", line 139, in <module>
y = model(img)  # dry run
  File "/home/zhangcheng/anaconda3/envs/rknn/lib/python3.6/site-packages/torch/nn/modules/module.py", line 727, in _call_impl
result = self.forward(*input, **kwargs)
  File "./models/common_rk_plug_in.py", line 122, in forward
x = self.conv1(x)
  File "/home/zhangcheng/anaconda3/envs/rknn/lib/python3.6/site-packages/torch/nn/modules/module.py", line 727, in _call_impl
result = self.forward(*input, **kwargs)
  File "/home/zhangcheng/anaconda3/envs/rknn/lib/python3.6/site-packages/torch/nn/modules/conv.py", line 423, in forward
return self._conv_forward(input, self.weight)
  File "/home/zhangcheng/anaconda3/envs/rknn/lib/python3.6/site-packages/torch/nn/modules/conv.py", line 420, in _conv_forward
self.padding, self.dilation, self.groups)
RuntimeError: Given groups=1, weight of size [3, 3, 1, 1], expected input[1, 640, 3, 640] to have 3 channels, but got 640 channels instead

```

作者: shaojia 时间: 2021-7-23 16:35
请问rk3399pro是不是不能跑YOLOV5?

欢迎光临 Toybrick (https://t.rock-chips.com/)