OpenCV: 在GPU端裁剪图像

Question

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

- Marek

Answer 1

经过对UMat类源代码和测试的研究，我找到了这个解决方案：

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

- Marek

---

在OpenCV 4.0.0中，我得到了以下错误：clip0 = cv2.UMat(umat0, [win_i, win_i+steps[0]], [win_j,win_j+steps[1]]) TypeError: UMat()最多接受2个参数（给出了3个）。 - Assaf-ge

Answer 2

def crop_umat(umat, height, width, left=0, top=0):
    return cv.UMat(umat, [top, height+top], [left, width+left])

- Petr Tripolsky