如何在带有cuda支持的python中使用opencv？

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

Full build information

如何解决？

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)