体验一下支持cuda加速的opencv

科幻木影

已于 2024-07-17 00:56:45 修改

阅读量1.3k

点赞数 5

分类专栏：妙算入坑文章标签： opencv 人工智能计算机视觉

于 2023-09-08 03:36:07 首次发布

本文链接：https://blog.csdn.net/ccs_jack/article/details/132751749

版权

妙算入坑专栏收录该内容

5 篇文章 0 订阅

订阅专栏

1、检查cuda设备是否可用

import cv2

print(cv2.cuda.getCudaEnabledDeviceCount())

cv2.cuda.printCudaDeviceInfo(0)

2、使用GPU模块

在python中，可以使用cv2.cuda模块来实现GPU加速，例如：

import cv2
import numpy as np

# 加载图像
img = cv2.imread('image.jpg', cv2.IMREAD_GRAYSCALE)

# 创建GPU图像对象
gpu_img = cv2.cuda_GpuMat()
gpu_img.upload(img)

# 创建Sobel算子
sobel = cv2.cuda.createSobelFilter(cv2.CV_8UC1, cv2.CV_8UC1, 1, 1)

# 运行Sobel算子
gpu_output = sobel.apply(gpu_img)

# 将输出从GPU对象中下载到CPU中
output = gpu_output.download()

# 显示输出
cv2.imshow('Output', output)
cv2.waitKey(0)

这里使用cv2.cuda_GpuMat()创建GPU图像对象，使用cv2.cuda.createSobelFilter()创建Sobel算子对象，使用sobel.apply()方法将算子应用于输入图像，最后使用gpu_output.download()方法将输出从GPU对象中下载到CPU中。

3、用python运行时的报错

可能会报Gtk-Message: 03:50:44.987: Failed to load module "canberra-gtk-module"

解决方法：

执行sudo apt-get install libcanberra-gtk-module安装即可

可能会报SyntaxError: Non-ASCII character '\xe5' in file cuda_test.py on line 3, but no encoding declared; see http://python.org/dev/peps/pep-0263/ for details

python默认编码是ASCII，报错信息的意思是找不到编码方式，也就是说出现了非法中文字符

解决方法：

在代码首行加上

# coding=UTF-8

或

# -*- coding:UTF-8 -*-

参考链接某乎