Toybrick

标题: rknn推理问题 [打印本页]

作者: Samuel 时间: 2019-10-2 15:34
标题: rknn推理问题
将转换得来的pb模型转换为rknn模型后，在rk3399pro上跑，输出的结果都是1，但是用pb模型预测是输出结果是正常的，模型转换的方法于WiKi教程里的一样。请问这是什么原因呢？

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

作者: xiaqing10 时间: 2019-10-10 13:40

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

一般是因为rgb转换的问题

作者: Samuel 时间: 2019-10-10 19:07

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

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

作者: Samuel 时间: 2019-10-12 17:29

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

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

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