Toybrick

楼主: Zen

yolov5的速度测试(更新原版yolov5结构导出)

chengzhang

新手上路

积分
44
11#
发表于 2021-5-11 14:32:42 | 只看该作者
```
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

新手上路

积分
32
12#
发表于 2021-7-23 16:35:05 | 只看该作者
请问rk3399pro是不是不能跑YOLOV5?
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

产品中心 购买渠道 开源社区 Wiki教程 资料下载 关于Toybrick


快速回复 返回顶部 返回列表