how opencv_traincascade can using GPU

asked 2017-12-05 23:58:24 -0600

Derick
1 ●1 ●2 ●3

I have try to train my classifier using opencv_traincascade in my machine. When run it, it utilize 100% of my CPU but not use my GPU. I have install OpenCV 3.x in my Ubuntu 16.04. And I have GeForece GTX 1080 Ti/PCIe/SSE2. I success install the driver with CUDA 8.0. How I can use the GPU instead of using CPU? I use below script to train the module in terminal

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

Any configuration I need to set to use the GPU?

edit retag flag offensive close merge delete

Comments

If you have a GTX1080Ti then skip the idea of cascade classifier all together and go for deep learning. Tons of efficient deep learning frameworks for object detection freely available.

StevenPuttemans ( 2017-12-06 04:27:07 -0600 )edit

add a comment

1 answer

Sort by » oldest newest most voted

answered 2017-12-06 02:11:37 -0600

berak
32993 ●7 ●81 ●312

sadly, there is no gpu code for the cascade training at all (it's not a configuration problem)

also, while there is cuda optimized detection code , you'd have to use the "old format" style cascades, generated from the opencv_haartraining tool (which is only in the outdated 2.4 branch)