将图像裁剪为 3d 数组

Question

我有一个尺寸为(224, 224, 3) 的tensor 代表图像。我想首先使用bndbox = [x1, y1, x2, y2] 格式的边界框尺寸“裁剪”此图像，然后将此裁剪后的图像调整回(224, 224, 3) 的尺寸。

有没有使用 numpy/cv2(OpenCV) 的简单方法？

Answer 1

假设你的张量支持切片，只需选择边界框，然后你可以用cv2.resize调整大小：

cv2.resize(img[y1:y2,x1:x2], (224, 224))

请注意，img 可能先选择高度然后选择宽度，而 cv2.resize 的形状参数采用 (width, height)。