The output of the object category detected by onnxruntime C++ is all 0

[GitHubLuGeng]

🐛 Describe the bug

onnxruntime C++检测到的物体输出类型全是person
相同yolov5s v6.0 onnx模型python demo推理正确

如下图：

Versions

main

[zhiqwang]

Hi @GitHubLuGeng

boxes 信息似乎都是对的哈？ person是 COCO 的第一类，似乎把 labels 都算成 0 了。

周末我实测一下看看问题是哪里导致的。

[zhiqwang]

Hi @GitHubLuGeng

The reason for this occurs should be a bug of the type casting of the following classId, it should be int64_t rather than float, we fixed this issue in #221 .

https://github.com/zhiqwang/yolov5-rt-stack/blob/b8d845d1c164d86723e79ed4ad0a288864becc12/deployment/onnxruntime/main.cpp#L150

Please pull the latest code to verify that this issue is fixed correctly.

And as such I'm closing this issue, feel free to reopen this or create a new ticket if you have more questions.

cc @itsnine @xiguadong .