rknn推理问题

[复制链接] · 发表于 2019-10-2 15:34:43

将转换得来的pb模型转换为rknn模型后，在rk3399pro上跑，输出的结果都是1，但是用pb模型预测是输出结果是正常的，模型转换的方法于WiKi教程里的一样。请问这是什么原因呢？

只看该作者 · 发表于 2019-10-8 18:21:04

求个大佬帮忙看看，，训练模型用的是tf1.10，开发板上也装的是tf1.10，固件是刷的V1.5版的

只看该作者 · 发表于 2019-10-10 13:40:18

Samuel 发表于 2019-10-8 18:21
求个大佬帮忙看看，，训练模型用的是tf1.10，开发板上也装的是tf1.10，固件是刷的V1.5版的 ...

一般是因为rgb转换的问题

只看该作者 · 发表于 2019-10-10 19:07:30

xiaqing10 发表于 2019-10-10 13:40
一般是因为rgb转换的问题

额，，我用cv2读取图片后再用cv2将图片通道转换为rgb格式，网络也是用的rgb格式的图片训练的，模型推理代码和模型转换代码如下。如此得到的模型的预测结果还是都是1.

只看该作者 · 发表于 2019-10-12 17:29:35

Samuel 发表于 2019-10-10 19:07
额，，我用cv2读取图片后再用cv2将图片通道转换为rgb格式，网络也是用的rgb格式的图片训练的，模型推理代 ...

从模型转换的log来看也没什么问题，，不知道问题出在哪