提问人:user3116857 提问时间:11/11/2023 最后编辑:Christoph Rackwitzuser3116857 更新时间:11/12/2023 访问量:27
无框坐标 OpenCV::d nn 推理 GPU
No boxes coordinates OpenCV::dnn inference GPU
问:
我正在运行 YOLOv8 的推理 (C++) 示例,您可以在官方存储库中找到该示例。它在 CPU 上工作正常(速度慢),但它正确地绘制了框,没有使用 GPU,它保持在 0%。
当我将其更改为 GPU 时,它不会绘制盒子。我可以看到 GPU 在处理图片所需的时间内以 25%(它是 NVIDIA 4080)运行。
我开始调试代码,我可以看到它检测到具有 GPU 和 CPU 的相同数量的盒子,但 GPU 模式下的坐标始终为 0。
OpenCV 是在 CUDA 和 CUDNN 支持下编译的,我可以看到 opencv 检测到我的 nvidia 卡。
知道为什么 OpenCV 不返回 GPU 上的框坐标吗?
问候
我期望在 GPU 或 CPU 中出现相同的行为,只是在 GPU 上更快。
答:
1赞
user3116857
11/12/2023
#1
经过一整天的搜索,我找到了解决方案。
https://github.com/ultralytics/ultralytics/issues/3682
就我而言,我只需要从 OpenCV 4.8 回滚到 OpenCV 4.7。经过测试并正常工作。
我希望它有所帮助。
评论
0赞
Zrufy
12/15/2023
这是唯一适合我的解决方案!
评论