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

[复制链接] · 发表于 2019-8-16 17:12:02

我从网上下载的pb，在调用rknn.load_tensorflow函数时，后面的inputs,outputs,input_size_list怎么从pb模型中获得啊？

只看该作者 · 发表于 2019-8-16 17:50:29

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