FaceNet的rknn模型推理结果和pb模型结果不同

显示全部楼层 · 发表于 2019-8-7 10:01:13

从代码里看，
1. 你的traning node没有关闭，推理时候引入training node是有可能造成结果随机的，特别是带dropout时候
2. inputoutput是代码里定义的，可以知道，不知道的话直接print 那层.name 就可以知道。也可以下载netron这个app来看模型，列出来的name是完整的
3. 推理出来数据不同可能原因就是同第一点，你的traning node没去掉，带有一些特殊训练的op。第二就是先关闭量化看看结果是否一致。第三，input数据的归一化设置是否正确

显示全部楼层 · 发表于 2019-8-8 13:56:19

pb用tensorflow载入跑过么？

显示全部楼层 · 发表于 2019-8-9 12:51:03

哦，刚看出来，你这里多输入了，目前rknn只支持单输入多输出。
所以你模型固化时候phase_train要固化成固定值，或者使用npz传入，具体可以看看文档。

显示全部楼层 · 发表于 2019-9-22 10:08:33

puyanan 发表于 2019-9-21 14:58
版主，您好。我发现phase_train的类型是placeholder，无法设置为固定值再固化，请问您是怎么做的呢？现在 ...

1.2版本rknn-toolkit允许多输入了，所以你可以不用固化placeholder