无框坐标 OpenCV::d nn 推理 GPU

No boxes coordinates OpenCV::dnn inference GPU

提问人:user3116857 提问时间:11/11/2023 最后编辑:Christoph Rackwitzuser3116857 更新时间:11/12/2023 访问量:27

问:

我正在运行 YOLOv8 的推理 (C++) 示例,您可以在官方存储库中找到该示例。它在 CPU 上工作正常(速度慢),但它正确地绘制了框,没有使用 GPU,它保持在 0%。

当我将其更改为 GPU 时,它不会绘制盒子。我可以看到 GPU 在处理图片所需的时间内以 25%(它是 NVIDIA 4080)运行。

我开始调试代码,我可以看到它检测到具有 GPU 和 CPU 的相同数量的盒子,但 GPU 模式下的坐标始终为 0。

OpenCV 是在 CUDA 和 CUDNN 支持下编译的,我可以看到 opencv 检测到我的 nvidia 卡。

知道为什么 OpenCV 不返回 GPU 上的框坐标吗?

问候

我期望在 GPU 或 CPU 中出现相同的行为,只是在 GPU 上更快。

C++ OpenCV 深度学习 Yolov8

评论


答:

1赞 user3116857 11/12/2023 #1

经过一整天的搜索,我找到了解决方案。

https://github.com/ultralytics/ultralytics/issues/3682

就我而言,我只需要从 OpenCV 4.8 回滚到 OpenCV 4.7。经过测试并正常工作。

我希望它有所帮助。

评论

0赞 Zrufy 12/15/2023
这是唯一适合我的解决方案!