Toybrick

标题: 如何把网上下载的pb转换为rknn [打印本页]

作者: jiajia1990 时间: 2019-8-16 17:12
标题: 如何把网上下载的pb转换为rknn
我从网上下载的pb，在调用rknn.load_tensorflow函数时，后面的inputs,outputs,input_size_list怎么从pb模型中获得啊？

作者: troy 时间: 2019-8-16 17:50
网上找下对应模型的TensorFlow的推理代码，推理代码里面自然会有这些信息。还有一个比较好用的工具网站是https://lutzroeder.github.io/netron/。可以查看模型的节点和网络结构。

欢迎光临 Toybrick (https://t.rock-chips.com/)