从 bbox 保存裁剪的图像

Question

我正在使用 cvlib 来检测 object 并且我希望能够保存基于 bbox 坐标的裁剪图像。

我的代码中有这个：

 def detect_object(img):
       
    # Open image
    image_stream = io.BytesIO(img)
    image_stream.seek(0)
    file_bytes = np.asarray(bytearray(image_stream.read()), dtype=np.uint8)
    frame = cv2.imdecode(file_bytes, cv2.IMREAD_COLOR)

    # Detection
    bbox, label, conf = cv.detect_common_objects(frame)
    output_image = draw_bbox(frame, bbox, label, conf)

    return output_image, bbox, label, conf

当我打印 bbox y 得到：

[3, -23, 1231, 731]

所以，我想使用这些坐标来裁剪原始图像并只保存这些坐标定义的检测到的 object

像这样的东西：

crop = output_image[bbox[2]:bbox[1], bbox[0]:bbox[3],:]
cv2.imwrite("crop.png", crop)

但是当我这样做时，我意识到作物不包含所需的 object，坐标是错误的。

我该如何解决？ 为什么我得到负坐标？

我的图像是 1280x720，所需的 object 几乎占据了整个图像。

Answer 1

我使用@Christoph Rackwitz 的评论作为参考解决了它，即处理偏移量：

image_height, image_width, image_channels = output_image.shape
box_xmin, box_ymin, box_xmax, box_ymax = bbox

if box_xmin < 0:
    box_xmin = 0
                    
if box_ymin < 0:
    box_ymin = 0

if box_xmax > image_width:
    box_xmax = image_width

if box_ymax > image_height:
    box_ymax = image_height

                    
crop = output_image[box_ymin:box_ymax, box_xmin:box_xmax,:]
cv2.imwrite("crop.jpg", crop)