ffmedia推理模块如何设置outputs[0].want_float = 1

nboxcn · 发表于 2024-2-26 10:27:09

使用推理模块 ModuleInference，有些模型需要输出数据为float，如何在对推理的输出数据格式进行设置为类似outputs[0].want_float = 1的方式？谢谢！

dengkx · 发表于 2024-2-27 15:24:23

本帖最后由 dengkx 于 2024-2-27 15:54 编辑

在setModelData()添加模型之后和在模块init()之前，可以更改：
```
output_attrs  = *(getOutputAttrPtr());
for (auto it : output_attrs) {
      it->type = RKNN_TENSOR_FLOAT32;
}
```

改变了数据类型或许也需要改变size?
```
   it->size = it->n_elems * sizeof(float);

```

nboxcn · 发表于 2024-2-27 16:06:20

感谢大神指点，晚点测试一下

nboxcn · 发表于 2024-2-27 16:52:48

是否可以开源一两个模块，方便我们进行参考扩展，例如开源：
1. vp的1个module_inference 或者module_rga
2. vo的1个module_fileWriter
或者把文档更完善一下，指导开发者如何扩展vp/vo模块

[Linux] ffmedia推理模块如何设置outputs[0].want_float = 1

【Linux】 ffmedia推理模块如何设置outputs[0].want_float = 1