Toybrick

tensorflow keras模型转rknn报错,如何解决?

wen0554

新手上路

积分
26
楼主
发表于 2020-9-25 15:40:48 | 显示全部楼层
你的参数写错了,应该是:
rknn.load_keras(model=H5_PATH, inputs='input_1_1', input_size_list='416,416,3', outputs='conv2d_59_1/BiasAdd, conv2d_67_1/BiasAdd, conv2d_75_1/BiasAdd')

另外修改一下
lib/python3.6/site-packages/rknn/ap/rknn.py
回复

使用道具 举报

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

本版积分规则

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


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