ckpt模型转换成pb再转换成rknn再rk3399pro上部署

[复制链接] · 发表于 2019-4-16 21:19:33

我想在板子上部署个单目深度计算的，用的NYU的模型，但其只提供了ckpt的模型，转换成pb和rknn的模型怎么弄呢？我始终找不到输入和输出节点的名称，能哪位大神帮忙看看吗？感激。
NYU的单目视觉的GitHub地址： https://github.com/iro-cp/FCRN-DepthPrediction

只看该作者 · 发表于 2019-4-18 09:36:43

用Netron可以看到各个节点真实名字

只看该作者 · 发表于 2020-6-11 10:36:37

jefferyzhang 发表于 2019-4-18 09:36
用Netron可以看到各个节点真实名字

pytorch生成的pt模型可以转成rknn么，v1.13.0

只看该作者 · 发表于 2020-6-11 10:38:05

zwj1234 发表于 2020-6-11 10:36
pytorch生成的pt模型可以转成rknn么，v1.13.0

看rknn-toolkit文档。1.3以后就可以了，但是pytorch只支持到1.2，
仔细看文档，并使用最新rknntoolkit