Toybrick

pb 轉 rknn 失敗

jefferyzhang

版主

积分
13005
楼主
发表于 2020-3-30 16:58:36 | 显示全部楼层
单从log上看,是 E Unknow layer "tensorarraygatherv3"  这个OP不支持。

这个问题很微妙,我自己遇到过这类问题,可能性和解决方法大概有以下几种:

1. 一种是用了keras.backend.set_learning_phase(0)在session创建之前,然后freeze pb,就饶过了这个op
2. 一种是尝试降低了tf的版本,因为TensorArrayGatherV3这个op不支持,但是TensorArrayGatherV2、TensorArrayGatherV1可能是支持的
3. 如果你是在尝试faster/mask r-cnn,这个op绕不过去,tf版本无法使用,可以尝试更换框架,用caffe的模型或者onnx的模型试试
4. 没有其他办法的时候,只能尝试自定义OP,可以参看我们的自定义OP文档自己实现这个op
回复

使用道具 举报

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

本版积分规则

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


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