opencv_traincascade使用GPU的列车分类器

时间:2017-12-06 08:19:21

标签: gpu opencv3.0 cascade-classifier

我尝试在我的机器中使用opencv_traincascade训练我的分类器。运行时,它使用100%的CPU,但不使用我的GPU。我在我的Ubuntu 16.04中安装了OpenCV 3.x.我有GeForece GTX 1080 Ti / PCIe / SSE2。我成功安装了CUDA 8.0的驱动程序。我如何使用GPU而不是使用CPU?我使用下面的脚本来训练模块

opnencv_traincascade -data data -vec positives.vec -bg bg.txt -numPos 37000 -numNeg 756 -numStage 20 -w 20 -h 20

我需要设置哪些配置才能使用GPU?

1 个答案:

答案 0 :(得分:1)

要将OpenCV与多线程和CUDA一起使用,您需要构建一个OpenCV,如此guide

或者你可以download已经组装好的OpenCV,首先安装所有必要的组件(CUDA Toolkit,TBB,MKL)并在Path

中设置环境变量