yolo模型在RK3399Pro部署的相关问题

[复制链接] · 发表于 2019-10-22 13:04:22

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

只看该作者 · 发表于 2019-10-22 15:19:26

本帖最后由 kepurSong 于 2019-10-22 20:19 编辑

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

yolo模型在RK3399Pro部署的相关问题

本帖子中包含更多资源

本帖子中包含更多资源