Как выполнить скрипт Python (распознавание лиц на очень большом наборе данных) на графическом процессоре Nvidia - PullRequest
0 голосов
/ 08 октября 2019

У меня есть скрипт Python, который просматривает набор видео и применяет функцию детектора лица и губ к каждому видео. Функция возвращает трехмерный массив данных пикселей, сосредоточенных на человеческих губах в каждом кадре.

Набор данных довольно большой (всего 70 ГБ, ~ 500 000 видео каждые около 1 секунды по продолжительности), и выполнение на обычном ЦП займет дни. У меня есть Nvidia 2080 Ti, который я хотел бы использовать для выполнения кода. Можно ли включить некоторый код, который выполняет весь мой сценарий на доступном графическом процессоре? Или я слишком упрощаю сложную проблему?

До сих пор я пытался реализовать с использованием numba и pycuda и не добился никакого прогресса, поскольку представленные примеры не очень хорошо соответствуют моей проблеме.

...