Как использовать OpenCV с питоном поддержки CUDA?

Question

В официальном репозитории opencv github я нашел пример ускоренного кода cuda для opencv. test_cuda.py

cuMat1 = cv.cuda_GpuMat()
cuMat2 = cv.cuda_GpuMat()
cuMat1.upload(npMat1)
cuMat2.upload(npMat2)

cuMat1 = cv.cuda.cvtColor(cuMat1, cv.COLOR_RGB2GRAY)
cuMat2 = cv.cuda.cvtColor(cuMat2, cv.COLOR_RGB2GRAY)

Но я обнаружил, что модуль 'cv2.cuda' не имеет атрибута 'cvtColor'.

Моя информация о сборке opencv:

NVIDIA CUDA:                   YES (ver 10.0, CUFFT CUBLAS FAST_MATH)
NVIDIA GPU arch:             61
NVIDIA PTX archs:            61

Полная информация о сборке

Как это исправить?

Answer 1

Вы должны измениться так.

cuMat1 = cv.cuda_GpuMat()
cuMat2 = cv.cuda_GpuMat()

npMat1_n = cv.cvtColor(npMat1, cv.COLOR_RGB2GRAY)
npMat2_n = cv.cvtColor(npMat2, cv.COLOR_RGB2GRAY)

cuMat1.upload(npMat1_n)
cuMat2.upload(npMat2_n)