RKNN量化失败无法匹配TensorArrayGatherV3

[复制链接] · 发表于 2021-3-1 15:44:58

TensorArrayGatherV3是tensorflow近几年来所有主流模型的张量处理op rknn不支持该op是要彻底放弃tensorflow了吗？
难受的是网上搜索不到这个op的有关信息因为无法自定义op 看名字像是一个张量拼接的op 有人成功量化过tensorflow object detection api的模型吗？

只看该作者 · 发表于 2021-3-1 15:58:14

1. 查看最新版本rknn-toolkit的OPList，没有的话就是暂时不支持的。
2. 一般会把输出节点定在tensorarraygatherv3之前一个节点，tensorarraygatherv3放上位机cpu做。
3. 也可以通过rknn的自定义OP自行增加该op支持(自定义op仅支持tf版本模型)

RKNN量化失败 无法匹配TensorArrayGatherV3

RKNN量化失败无法匹配TensorArrayGatherV3