Toybrick

---onnx模型转换错误!

linuxdog

注册会员

积分
82
楼主
发表于 2022-3-7 22:46:16    查看: 3052|回复: 1 | [复制链接]    打印 | 只看该作者
我用如下网站下的代码,做了pretrain和提取了onnx模型文件
https://github.com/ultralytics/yolov5

然后放在了:  /rknn-toolkit2-1.0.0_no_docker/examples/onnx/bztest
目录下做了转换处理: (bztest目录是我拷贝resnet50v2的副本改出来的)
(原本的resnet50v2/这个目录下跑python3做模型转换是没有问题的)

$ python3 test.py 跑出来如下错误结果,请问这是啥问题呢?

--> config model
done
--> Loading model
done
--> Building model
E build: Catch exception when building RKNN model!
E build: Traceback (most recent call last):
E build:   File "rknn/api/rknn_base.py", line 2140, in rknn.api.rknn_base.RKNNBase.build
E build:   File "rknn/api/graph_optimizer.py", line 2297, in rknn.api.graph_optimizer.GraphOptimizer.run
E build:   File "rknn/api/graph_optimizer.py", line 2272, in rknn.api.graph_optimizer.GraphOptimizer._sparse_weight
E build:   File "rknn/api/sparse_weight.py", line 521, in rknn.api.sparse_weight.sparse_weight
E build:   File "rknn/api/sparse_weight.py", line 444, in rknn.api.sparse_weight._range_add
E build: ValueError: operands could not be broadcast together with shapes (80,) (3,)
Build resnet50 failed!


回复

使用道具 举报

linuxdog

注册会员

积分
82
沙发
 楼主| 发表于 2022-3-8 10:09:57 | 只看该作者
太安静了不是
回复

使用道具 举报

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

本版积分规则

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


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