Toybrick

模型转换问题

czcl

中级会员

积分
383
楼主
发表于 2022-9-20 16:21:23    查看: 2292|回复: 0 | [复制链接]    打印 | 只看该作者
使用darknet 训练的分类模型,然后使用官方代码转换rknn模型报错,使用训练的检测网络(yolov系列) 没有问题


平台:rk3588,PC模拟环境下转换

W __init__: rknn-toolkit2 version: 1.3.0-11912b58
--> Loading model
Activation not supported.
Activation not supported.
E load_darknet: Catch exception when loading darknet model !
E load_darknet: Traceback (most recent call last):
E load_darknet:   File "rknn/api/rknn_base.py", line 1507, in rknn.api.rknn_base.RKNNBase.load_darknet
E load_darknet:   File "rknn/base/convertor/yolo2onnx/util/onnx_util.py", line 366, in rknn.base.convertor.yolo2onnx.util.onnx_util.GraphBuilderONNX.build_onnx_graph
E load_darknet:   File "rknn/base/convertor/yolo2onnx/util/onnx_util.py", line 445, in rknn.base.convertor.yolo2onnx.util.onnx_util.GraphBuilderONNX._make_onnx_node
E load_darknet:   File "rknn/base/convertor/yolo2onnx/util/onnx_util.py", line 775, in rknn.base.convertor.yolo2onnx.util.onnx_util.GraphBuilderONNX._make_maxpool_node
E load_darknet: KeyError: 'layer_dict'
load darknet model failed!

1 请问应该怎么处理
2 有没有办法导出onnx的模型


回复

使用道具 举报

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

本版积分规则

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


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