Классификатор поездов opencv_traincascade с использованием графического процессора

Я пытаюсь обучить свой классификатор с помощью opencv_traincascade на моей машине. При запуске он использует 100% моего процессора, но не использует мой графический процессор. Я установил OpenCV 3.x в свой Ubuntu 16.04. А у меня GeForece GTX 1080 Ti / PCIe / SSE2. Я успешно установил драйвер с CUDA 8.0. Как я могу использовать графический процессор вместо процессора? Я использую ниже скрипт для обучения модуля в

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

Любая конфигурация, которую мне нужно настроить для использования графического процессора?


person Bubble Bub    schedule 06.12.2017    source источник


Ответы (1)


arrow_upward
1
arrow_downward

Чтобы использовать OpenCV с многопоточностью и CUDA, вам необходимо создать OpenCV, как в этом руководство

Или вы можете загрузить уже собранный OpenCV, предварительно установив все необходимые компоненты (CUDA Toolkit, TBB, MKL) и установка переменных среды в Path

person Денис Бабушкин    schedule 15.03.2018
comment
Очень полезное руководство. - person FindOutIslamNow; 05.11.2018