c++ 无框坐标OpenCV::dnn推理GPU

hgqdbh6s  于 11个月前  发布在  其他
关注(0)|答案(1)|浏览(103)

我运行的例子推理(C++)为YOLOv 8,你可以在官方仓库中找到.它在CPU上工作正常(慢),但它画正确的盒子,没有使用GPU,它保持在0%.
当我把它换成GPU时,它不会画盒子。我可以看到GPU在处理图片所需的时间内以25%的速度运行(它是NVIDIA 4080)。
我开始调试代码,我可以看到它检测到GPU和CPU相同数量的盒子,但GPU模式下的坐标始终为0。
OpenCV是用CUDA和CUDNN支持编译的,我可以看到OpenCV检测到我的NVIDIA卡。
你知道为什么OpenCV不返回GPU上的盒子坐标吗?
问候
我期待在GPU或CPU中有相同的行为,只是在GPU上更快。

q5lcpyga

q5lcpyga1#

经过一整天的寻找,我找到了解决办法。
https://github.com/ultralytics/ultralytics/issues/3682
在我的情况下,我只需要从OpenCV 4.8回滚到OpenCV 4.7。测试和工作。
希望能帮上忙。

相关问题