单层卷积计算转换rknn模型

[复制链接] · 发表于 2019-9-5 15:11:10

你好
我现在需要将一个只有一层卷积计算的pb模型转换成rknn模型，其中卷积核大小为5*5，深度为1，参数全为1。在转换完rknn模型后，我用一个680*480*1的一矩阵作为输入，计算后结果很奇怪，参数并不是25，请问是在转换模型时出了什么问题吗？

谢谢！

只看该作者 · 发表于 2019-9-5 18:17:47

rknn-toolkit 升级到1.2.0后再试，1.1在单层卷积上有一个bug。
今天刚帮客户解决了一个单层卷积输出的问题。
然后输入记得要强制指定类型和rknn的input一致，不然你输入float64的array，到rknn按float32解析就全错了

只看该作者 · 发表于 2019-9-6 08:42:29

好的非常感谢！