OpenCV使用CUDA处理图像的教程与实战

import cv2 as cv

gpu_frame = cv.cuda_GpuMat()

screenshot = cv.imread('media/drip.png')

gpu_frame.upload(screenshot)

gpu_frame.download()

概述

• 在单张图像上使用

• 在多张图像上使用

• 对多张图像使用Dask进行并行延时处理

在单张图像上使用

我们需要创建GPU空间(GPU_frame)来保存图像(就像相框保存图片一样)，然后才能将图像上传到GPU。

第1步：上传

import cv2 as cv

gpu_frame = cv.cuda_GpuMat()

接下来用CPU将图像加载到内存中(截图)，并将其上传到gpu上(帧图像)；

screenshot = cv.imread('media/drop.png')

gpu_frame.upload(screenshot)

第2步：处理图像

OpenCV CUDA函数返回cv2.cuda_GpuMat(GPU矩阵)，因此每个结果都可以在用户不必重新上传的情况下进行操作。

让我们把图像从RGB转换成B