Toybrick

标题: yolo模型在RK3399Pro部署的相关问题 [打印本页]

作者: kepurSong 时间: 2019-10-22 13:04
标题: yolo模型在RK3399Pro部署的相关问题
1，我们这边是用keras版本的yolov3进行物体检测的，生成的权重是.h5格式；
2，我们这里是先将.h5转化为.pb格式，这个过程没有问题
3，我们在将.pb转化为.rknn的时候，报错了，提示是：不支持LeakyRelu
报错的.log文件以及相关文件-->>见：附件1（这里我output层改为第一次用到leakrule之后的Conv2D那一层）

作者: kepurSong 时间: 2019-10-22 15:19
本帖最后由 kepurSong 于 2019-10-22 20:19 编辑

我去除了leakrelu就可以了，
但是，最终我用yolov3.rknn推理的图片是一个（1， 1， 80640），不是三个的，不能直接用官网的代码，这个是什么原因？
推理出来的shape不一样，这里我自己尝试分解了一下，无头可寻。。。，有点麻烦。图像显示如下（其中一个乱码）：

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