Алгоритмы сегментации

[maxfashko 5]

Здравствуйте!

Интересует собственно, какие вы знаете алгоритмы сегментации реализованные с поддержкой вычислений на gpu.

А задача такая - имеется исходное изображение ( rgb ), имеется маска при помощи которой необходимо "отсегментировать" исходное изображение ( 1 - объект, 0 - фон )

В данный момент использую алгоритм grabcut реализованный в opencv. Справляется с поставленной задачей очень-очень хорошо, но есть недостатки:

Для качественной сегментации одного изображения требуется 2 запуска алгоритма. 

1 - первичное уточнение прямоугольной областью

2 - Удаление фона

Так как в opencv grabcut вычисляет все на cpu, получается ну очень долго ( 4 - 5 секунд: img 1500*2500px )

Nvidia реализовала очень давно grabcut, и запихнула все это дело в примеры cuda. Но так как cuda обновилась уже несколько раз, этот код не рабочий на новых версиях cuda.

[Smorodov 579]

DenseCut : http://mmcheng.net/densecut/

Есть еще любопытная библиотека: http://research.project-10.de/dgm/doc/index.html  , но я пока не тестил.

Вроде в OpenCV была GPU-шная версия gpubcut, или ее тоже удалили ?

[maxfashko 5]

19 часов назад, Smorodov сказал:

Вроде в OpenCV была GPU-шная версия gpubcut, или ее тоже удалили ?

Вообще не припомню чтобы когда-то там в примерах была версия для gpu

[Smorodov 579]

Ну наверное я перепутал ее с той что в примерах к CUDA была.