模型转换问题

[复制链接] · 发表于 2022-9-20 16:21:23

使用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的模型