DeepStream-Yolo项目中如何将模型预测结果保存到文本文件

在基于DeepStream框架的视频分析应用中，我们经常需要将模型检测结果保存下来用于后续分析。本文将详细介绍在DeepStream-Yolo项目中如何配置将YOLO模型的预测结果输出到文本文件的方法。

配置方法

在DeepStream应用配置文件中，我们可以通过设置gie-kitti-output-dir参数来指定输出目录，系统会自动将检测结果保存为KITTI格式的文本文件。KITTI格式是一种常用的目标检测数据格式，每行包含一个检测对象的信息。

参数详解

gie-kitti-output-dir参数需要设置在GIE(推理引擎)配置部分，例如：

[primary-gie]
enable=1
gpu-id=0
gie-unique-id=1
nvbuf-memory-type=0
config-file=config_infer_primary_yoloV8.txt
gie-kitti-output-dir=/path/to/output/directory

输出文件格式

系统会为每个视频帧生成一个对应的文本文件，文件内容包含以下信息：

• 检测类别名称
• 置信度分数
• 边界框坐标(x1, y1, x2, y2)

这种格式非常适合后续的数据分析和处理，可以直接被大多数机器学习工具读取和使用。

多模型输出处理

对于项目中使用的多级模型架构(如主模型和次级模型)，可以为每个GIE模块单独设置输出目录，这样就能分别保存不同模型的检测结果。

性能考虑

需要注意的是，开启文本输出会增加一定的IO开销，在性能敏感的场景下需要评估其对整体处理速度的影响。建议在开发调试阶段开启此功能，而在生产环境中根据实际需求决定是否启用。

通过这种简单的配置，开发者可以方便地获取模型推理结果，为进一步的分析和处理提供了基础数据支持。