Toybrick

标题: onnx转换成rknn格式出错 [打印本页]

作者: btxc_cyy 时间: 2022-3-21 10:08
标题: onnx转换成rknn格式出错
根据这个git里的代码littledeep/YOLOv5-RK3399Pro: PyTorch-->ONNX-->RKNN (github.com)执行convert/rknn_convert.py进行onnx->rknn转换的时候，报了图片里的错误。

ONNX的模型是可以正常进行推理的。
请问各位有遇到类似的问题吗？

作者: hbsyxyt 时间: 2022-3-22 09:14
我也遇到了这个问题，请问楼主解决了吗？

作者: btxc_cyy 时间: 2022-3-22 10:42

hbsyxyt 发表于 2022-3-22 09:14
我也遇到了这个问题，请问楼主解决了吗？

问题本身没解决，但是发现用官方的YOLOV5里的export导出的onnx是不会出现这个问题的。

作者: hbsyxyt 时间: 2022-3-23 10:43

btxc_cyy 发表于 2022-3-22 10:42
问题本身没解决，但是发现用官方的YOLOV5里的export导出的onnx是不会出现这个问题的。 ...

好的我试一下，谢谢

作者: tiger_k 时间: 2022-3-24 14:47

btxc_cyy 发表于 2022-3-22 10:42
问题本身没解决，但是发现用官方的YOLOV5里的export导出的onnx是不会出现这个问题的。 ...

用yolov5官方的代码导出的话导出命令是要变吗？
是这个吗

python export.py --weights yolov5s.pt --img 640 --batch 1 --opset 12
复制代码

楼主用的是哪个版本的yolov5呢

作者: btxc_cyy 时间: 2022-3-25 17:44

tiger_k 发表于 2022-3-24 14:47
用yolov5官方的代码导出的话导出命令是要变吗？
是这个吗

官方的example里有个readme，里面有commit id，用这个版本的就可以， export的命令一样

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