为了加速一些操作,我将图像传输到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内存中有一种裁剪图像的方法。有什么提示吗?