python - opencv:在 GPU 端裁剪图像

Question

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

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

问题是无法在 GPU 内存中对图像进行切片。上面的代码会引发错误:“UMat object is not subscriptable”。我找到的解决方案是将图像传回 numpy 对象，然后将其切片，然后传回 GPU:

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

但上述解决方案看起来成本很高，因为将数据传输到 GPU 内存或从中传输数据需要时间。我感觉有一种方法可以在 GPU 内存中裁剪图像。有什么提示吗？

Answer 1

在研究了 UMat 类源代码和测试之后，我找到了这个解决方案:

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