Toybrick

标题: rknn模型输出节点名字设定 [打印本页]

作者: xkh_link 时间: 2023-4-21 16:56
标题: rknn模型输出节点名字设定
当使用rknn-toolkit工具时，生成的rknn模型输出节点的名字是以output开始的名字，请问可以指定生成的rknn模型输出节点的名字吗？如果可以的话，应该在哪里指定呢？

作者: jefferyzhang 时间: 2023-4-22 08:58
可以，rknntoolkit工具文档里有介绍

作者: xkh_link 时间: 2023-4-23 09:13

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

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

作者: jefferyzhang 时间: 2023-4-23 10:31

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

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

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