Toybrick

如何把网上下载的pb转换为rknn

jiajia1990

中级会员

积分
448
发表于 2019-8-16 17:12:02    查看: 6159|回复: 1 | [复制链接]    打印 | 显示全部楼层
我从网上下载的pb,在调用rknn.load_tensorflow函数时,后面的inputs,outputs,input_size_list怎么从pb模型中获得啊?
回复

使用道具 举报

troy

版主

积分
2346
发表于 2019-8-16 17:50:29 | 显示全部楼层
网上找下对应模型的TensorFlow的推理代码,推理代码里面自然会有这些信息。还有一个比较好用的工具网站是https://lutzroeder.github.io/netron/。可以查看模型的节点和网络结构。
回复

使用道具 举报

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

本版积分规则

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


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