为 Windows 10 构建 OpenCV + CUDA + Contrib

如何解决为 Windows 10 构建 OpenCV + CUDA + Contrib

我正在尝试在 Python 的 OpenCV 中使用 cuda optical flow functions。我已经使用 cuda 成功构建了 OpenCV 4.5.2，并在 modules 的 opencv-contrib 中添加了 CMake 文件夹。当我想调用光流函数时，对于 Python 中的 cpu（例如 DualTVL1OpticalFlow）和 gpu（例如 OpticalFlowDual_TVL1），我得到以下错误：

• cpu：AttributeError: module "cv2" has not attribute optflow
• GPU：AttributeError: module "cv2" has not attribute cuda_NvidiaOpticalFlow_1_0

我很确定这与构建有关，因为当我使用 pip install opencv-python 和 pip install opencv-contrib-python 构建 opencv 时，我可以使用 cpu 函数，但是 我想从 Python 访问 gpu 函数. 我使用 cmake-gui 并启用以下标志：

• WITH_CUDA ☑
• OPENCV_DNN_CUDA ☑
• ENABLE_FAST_MATH ☑
• OPENCV_EXTRA_MODULES
• ENABLE_CUDA_FAST_MAT ☑
• CUDA_ARCH_BIN
• BUILD_opencv_cudaoptflow ☑
• BUILD_opencv_optflow ☑

我想知道能够从 Python 调用 cuda 光流函数的人是否可以帮助解决这个问题。谢谢。