Обрезка изображения в виде трехмерного массива

Question

У меня есть tensor размера (224, 224, 3), представляющего изображение. Я хотел бы сначала "обрезать" это изображение, используя размеры ограничительной рамки в формате 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 принимает (ширина, высота).