ffmedia推理模块如何设置outputs[0].want_float = 1
使用推理模块 ModuleInference,有些模型需要输出数据为float,如何在对推理的输出数据格式进行设置为类似outputs.want_float = 1的方式?谢谢!本帖最后由 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);
```
感谢 大神 指点,晚点测试一下 是否可以开源一两个模块,方便我们进行参考扩展,例如开源:
1. vp的1个module_inference 或者module_rga
2. vo的1个module_fileWriter
或者把文档更完善一下,指导开发者如何扩展vp/vo模块
页:
[1]