如何使用opencv與cuda支持python？

Question

我在官方opencv github存儲庫中找到了cuda加速opencv python代碼的示例。 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)