Toybrick

标题: RKNN模型输入输出index node [打印本页]

作者: eeyeh 时间: 2019-4-10 20:30
标题: RKNN模型输入输出index node
本帖最后由 eeyeh 于 2019-4-10 20:36 编辑

在看linux的demo rknn_ssd.cpp时, 输出部分output有两个
const int output_elems1 = NUM_RESULTS * 4;
const uint32_t output_size1 = output_elems1 * sizeof(float);
const int output_index1 = 0; // node name "concat"

const int output_elems2 = NUM_RESULTS * NUM_CLASSES;
const uint32_t output_size2 = output_elems2 * sizeof(float);
const int output_index2 = 1; // node name "concat_1"

。。。。。。
      ret = rknn_outputs_get(ctx, 2, outputs, nullptr);
      float* predictions = (float*)outputs[0].buf;
      float* outputClasses = (float*)outputs[1].buf;

似乎是通过index分别获得坐标和类别, 但模型训练的时候原来框架的output不是在一起的吗？
rknn在生成模型时候是如何将其分离成两个并分别赋上不同的index node？
看API文档中没有提到, 也没有找到生成模型的代码？
请问这部分在生成模型时候是怎么做的？有指导吗？

---------------------
好像tensorflow模型本身是可以多节点输出的
之前一直用caffe和darknet 不了解, 不好意思, 版主删了吧~

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