Toybrick

标题: YOLOv3 tensorflow版本 Pb转RKNN时报错 [打印本页]

作者: xiaobai 时间: 2019-11-26 15:28
标题: YOLOv3 tensorflow版本 Pb转RKNN时报错
本帖最后由 xiaobai 于 2019-11-26 15:41 编辑

使用rknn.load_tensorflow这个API时出现以下报错
AttributeError:'NoneType' object has no attribute 'lid;请问这个报错是什么意思？

作者: jefferyzhang 时间: 2019-11-27 08:28
虽然不知道你发的啥玩意儿，但是我猜过去3成是你pb文件有问题，4成是你生成pb的tf和转换rknn的tf版本不一致。
请先自己用当前tf读下pb推理看下是否正常

作者: xiaobai 时间: 2019-11-27 10:14

jefferyzhang 发表于 2019-11-27 08:28
虽然不知道你发的啥玩意儿，但是我猜过去3成是你pb文件有问题，4成是你生成pb的tf和转换rknn的tf版本不一致 ...

我自己的PC端读这个PB是可以正常推理的。我也考虑到可能tf版本的问题后来就直接在板子上把CKPT转PB了（PC的1.12tensorflow训练得到ckpt，传到板子上再ckpy转pb）然后还是报这个错。
请问这个和网络的输入大小有关系吗，因为训练的时候输入是几个尺度随机的，后来我改成只有416的还是不行。

作者: jefferyzhang 时间: 2019-11-27 10:51

xiaobai 发表于 2019-11-27 10:14
我自己的PC端读这个PB是可以正常推理的。我也考虑到可能tf版本的问题后来就直接在板子上把CKPT转PB了（PC ...

rknn必须固化输入尺寸。
PB转rknn你可以在pc上做，然后板子上load。
一般来说你pc仿真的rknn toolkit是可以读的话，同一个版本板子上也是没问题的。
当然pc仿真都无法进行，就是转换有问题了

作者: xiaobai 时间: 2019-11-30 16:41

jefferyzhang 发表于 2019-11-27 10:51
rknn必须固化输入尺寸。
PB转rknn你可以在pc上做，然后板子上load。
一般来说你pc仿真的rknn toolkit是可 ...

网络中有五维tensor可以吗

作者: jefferyzhang 时间: 2019-11-30 19:06

xiaobai 发表于 2019-11-30 16:41
网络中有五维tensor可以吗

网络内多少维理论上都可以，没有限制

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