mxnet insightface模型转rknn出错

[复制链接] · 发表于 2020-7-24 15:39:49

用到如下代码转换模型，出现错误，请问大神，是什么原因：
rknn = RKNN()
rknn.config(channel_mean_value='0 0 0 255', reorder_channel='0 1 2', target_platform='rk3399pro')
rknn.load_mxnet(
      symbol='model-symbol.json',
      params='model-0000.params',
      input_size_list=[[3, 112, 112]]
)
rknn.build(do_quantization=True, dataset='dataset.txt')

rknn.export_rknn('./r100_q.rknn')

# Release RKNN Context
rknn.release()

E Catch exception when loading mxnet model: /run/media/260260/db719e55-9988-4588-99dc-41b390376d40/pythonspace2/insightface_v2/models/model-r100-ii/model-symbol.json!
E Traceback (most recent call last):
E File "rknn/api/rknn_base.py", line 632, in rknn.api.rknn_base.RKNNBase.load_mxnet
E File "rknn/base/RKNNlib/app/importer/mxnetloader.py", line 570, in rknn.base.RKNNlib.app.importer.mxnetloader.MxnetLoader.load
E File "rknn/base/RKNNlib/app/importer/mxnetloader.py", line 461, in rknn.base.RKNNlib.app.importer.mxnetloader.MxnetLoader.parse_layers
E File "rknn/base/RKNNlib/app/importer/mxnetloader.py", line 369, in rknn.base.RKNNlib.app.importer.mxnetloader.MxnetLoader.parse_layer_params
E File "rknn/base/RKNNlib/app/importer/mxlayer_parse.py", line 224, in rknn.base.RKNNlib.app.importer.mxlayer_parse.parse_fullconnect_param
E KeyError: 'no_bias'

只看该作者 · 发表于 2021-9-21 20:46:37

楼主转成功了嘛？