rknn模型输出节点名字设定

[复制链接] · 发表于 2023-4-21 16:56:13

当使用rknn-toolkit工具时，生成的rknn模型输出节点的名字是以output开始的名字，请问可以指定生成的rknn模型输出节点的名字吗？如果可以的话，应该在哪里指定呢？

只看该作者 · 发表于 2023-4-22 08:58:55

可以，rknntoolkit工具文档里有介绍

只看该作者 · 发表于 2023-4-23 09:13:34

jefferyzhang 发表于 2023-4-22 08:58
可以，rknntoolkit工具文档里有介绍

请问在哪里指定呢？可以详细一些吗？我只在加载onnx模型的参数里找到了inputs和outputs参数，在API export_rknn没找到。

只看该作者 · 发表于 2023-4-23 10:31:43

xkh_link 发表于 2023-4-23 09:13
请问在哪里指定呢？可以详细一些吗？我只在加载onnx模型的参数里找到了inputs和outputs参数，在API expor ...

就是这个了，不然你以为加载的input和output是干嘛用的