Firefly开源社区

标题: ffmedia推理模块如何设置outputs[0].want_float = 1 [打印本页]

作者: nboxcn 时间: 2024-2-26 10:27
标题: ffmedia推理模块如何设置outputs[0].want_float = 1
使用推理模块 ModuleInference，有些模型需要输出数据为float，如何在对推理的输出数据格式进行设置为类似outputs[0].want_float = 1的方式？谢谢！

作者: dengkx 时间: 2024-2-27 15:24
本帖最后由 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
感谢大神指点，晚点测试一下

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

欢迎光临 Firefly开源社区 (https://dev.t-firefly.com/)