NVidia DeepStream 输出推理类不匹配 - “车辆类”

问题描述

• 硬件平台(Jetson / GPU) Jetson Nano 4GB,Ubuntu 18.4

• DeepStream 版本 marketplace.azurecr.io/nvidia/deepstream-iot2-l4t:latest

• JetPack 版本 4.3

• 问题类型 输出推理类与模型类不同

• 如何重现问题? 在 DeepStream 上,部署对象检测 ONNX 模型。我的模型是从 Azure 自定义视觉导出的 ONNX 模型。我的标签文件有 2 个类 - 'Mask'、'No_Mask'。部署工作正常,我能够使用 DeepStream 执行我的模型。但是,我得到的输出推理类为“Vehicle”和“No_Mask”。你能帮我理解为什么我的模型中不存在输出推理标签为“车辆”。

示例输出推理日志 {"log":" "1|324|23|380|61|车辆|#|||||||0"\n","stream":"stdout","time":"2021-01- 05T16:15:15.614591738Z"}

{"log":" "1|324|23|380|61|Vehicle|#|||||||0"\n","time":"2021 -01-05T16:15:15.614790179Z"}

{"log":" "2|141|15|365|161|No Mask"\n","time":"2021-01-05T16:15:15.614221209Z "}

解决方法

您很可能指定了错误的标签文件或其中的类是错误的。它在 labelfile-path 中提供为

labelfile-path=labels.txt