Toybrick

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

kepurSong

中级会员

积分
423
楼主
发表于 2019-10-22 13:04:22    查看: 8137|回复: 1 | [复制链接]    打印 | 只看该作者
    1,我们这边是用keras版本的yolov3进行物体检测的,生成的权重是.h5格式;
    2,我们这里是先将.h5转化为.pb格式,这个过程没有问题
    3,我们在将.pb转化为.rknn的时候,报错了,提示是:不支持LeakyRelu
    报错的.log文件以及相关文件-->>见:附件1(这里我output层改为第一次用到leakrule之后的Conv2D那一层)


本帖子中包含更多资源

您需要 登录 才可以下载或查看,没有帐号?立即注册

x
回复

使用道具 举报

kepurSong

中级会员

积分
423
沙发
 楼主| 发表于 2019-10-22 15:19:26 | 只看该作者
本帖最后由 kepurSong 于 2019-10-22 20:19 编辑

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



本帖子中包含更多资源

您需要 登录 才可以下载或查看,没有帐号?立即注册

x
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

产品中心 购买渠道 开源社区 Wiki教程 资料下载 关于Toybrick


快速回复 返回顶部 返回列表