python - 给定 4 点，如何从 pytorch/torchvision 中的图像中裁剪出四边形？

Question

问题很简单——

我想从 pytorch/torchvision 中的图像中裁剪出一个四边形。给定，我有这个四边形角的四个坐标。

请注意，这四个点在其内部限定了一个四边形，该四边形可能是矩形，也可能不是矩形。因此，请不要建议涉及图像切片的答案。

如果我错过了任何相关细节，请发表评论。

Answer 1

引用https://stackoverflow.com/a/30902423/4982729我可以使用 opencv 提取补丁 -

import numpy as np
import cv2

pts = np.array([[542, 107], [562, 102], [582, 110], [598, 142], [600, 192], [601, 225], [592, 261], [572, 263], [551, 245], [526, 220], [520, 188], [518, 152], [525, 127], [524, 107]], dtype=np.int32)

mask = np.zeros((img.shape[0], img.shape[1]))

cv2.fillConvexPoly(mask, pts, 1)
mask = mask.astype(np.bool)

out = np.zeros_like(img)
out[mask] = img[mask]

然后我可以手动将numpy数组转换为torch的变量。即使我正在为神经网络形成火炬图，这似乎也不会引发任何错误。

    outputs = model(images)
    o = outputs.data.cpu().numpy()
    #
    # do your opencv stuff here to o
    #
    o = torch.Tensor(o).to(device)
    outputs = Variable(o, requires_grad=True)