Toybrick

mxnet insightface模型转rknn出错

搬砖家

新手上路

积分
22
发表于 2020-7-24 15:39:49    查看: 7315|回复: 1 | [复制链接]    打印 | 显示全部楼层
用到如下代码转换模型,出现错误,请问大神,是什么原因:
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'

回复

使用道具 举报

柯南VS小智

新手上路

积分
14
发表于 2021-9-21 20:46:37 | 显示全部楼层
楼主转成功了嘛?
回复

使用道具 举报

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

本版积分规则

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


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