Toybrick

标题: ckpt模型转换成pb再转换成rknn再rk3399pro上部署 [打印本页]

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

作者: jefferyzhang 时间: 2019-4-18 09:36
用Netron可以看到各个节点真实名字

作者: zwj1234 时间: 2020-6-11 10:36

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

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

作者: jefferyzhang 时间: 2020-6-11 10:38

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

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

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