python – opencv：在GPU端裁剪图像

发布时间：2020-12-20 11:03:03 所属栏目：Python 来源：网络整理

导读：为了加快某些操作,我将图像传输到GPU内存.经过一些处理后,我必须裁剪一些区域： image = cv2.UMat(image)...crop = image[y1:y2,x1:x2] 问题是无法在GPU内存中切片图像.上面的代码会引发错误：“UMat对象不可订阅”.我找到的解决方案是将图像传输回numpy对象

为了加快某些操作,我将图像传输到GPU内存.经过一些处理后,我必须裁剪一些区域：

image = cv2.UMat(image)
...
crop = image[y1:y2,x1:x2]

问题是无法在GPU内存中切片图像.上面的代码会引发错误：“UMat对象不可订阅”.我找到的解决方案是将图像传输回numpy对象,而不是将其切片,然后传回GPU：

image = cv2.UMat(image)
...
image = image.get()
crop = image[y1:y2,x1:x2]
crop = cv2.UMat(crop)
...

但是上面的解决方案看起来很昂贵,因为向/从GPU内存传输数据需要时间.我觉得有一种方法可以在GPU内存中裁剪图像.任何提示？

解决方法

在对UMat类源代码和测试进行一些研究后,我找到了这个解决方案：

crop = cv2.UMat(image,[y0,y1],[x0,x1])

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!