Toybrick

标题: RKNN量化失败无法匹配TensorArrayGatherV3 [打印本页]

作者: 戈好雨 时间: 2021-3-1 15:44
标题: RKNN量化失败无法匹配TensorArrayGatherV3
TensorArrayGatherV3是tensorflow近几年来所有主流模型的张量处理op rknn不支持该op是要彻底放弃tensorflow了吗？
难受的是网上搜索不到这个op的有关信息因为无法自定义op 看名字像是一个张量拼接的op 有人成功量化过tensorflow object detection api的模型吗？

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

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