Toybrick

onnx模型转rk模型出错

weidao

新手上路

积分
29
发表于 2019-5-15 15:01:44    查看: 142|回复: 1 | [复制链接]    打印 | 显示全部楼层
如题,onnx模型是github上找的ArcFace模型(https://github.com/onnx/models/tree/master/models/face_recognition/ArcFace),加载报错。求解

代码如下:
import cv2
from rknn.api import RKNN

if __name__ == '__main__':

    # Create RKNN object
    rknn = RKNN()

    # pre-process config
    print('--> config model')
    rknn.config(channel_mean_value='0,0,0,255', reorder_channel='2 1 0')
    print('done')

    # Load tensorflow model
    print('--> Loading model')
    ret = rknn.load_onnx(model='./resnet100.onnx')
    if ret != 0:
        print('Load resnet100 failed! Ret = {}'.format(ret))
        exit(ret)
    print('done')

    # Build model
    print('--> Building model')
    ret = rknn.build(do_quantization=True, dataset='./dataset.txt')
    if ret != 0:
        print('Build resnet100 failed!')
        exit(ret)
    print('done')

    # Export rknn model
    print('--> Export RKNN model')
    ret = rknn.export_rknn('./resnet100.rknn')
    if ret != 0:
        print('Export mobilenet_v2.rknn failed!')
        exit(ret)
    print('done')

    rknn.release()


本帖子中包含更多资源

您需要 登录 才可以下载或查看,没有帐号?立即注册

x
回复

使用道具 举报

yaowei

注册会员

积分
143
发表于 2019-5-15 19:20:26 | 显示全部楼层
大佬我们是一样的错误
回复

使用道具 举报

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

本版积分规则

产品中心 购买渠道 开源社区 wiki教程 关于Toybrick


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