Function draw_rectangle

code_generation_benchmark/utils/utils_vis_bbox.py:6–21

Draw a rectangle on an image. Args: image: numpy array of shape (H, W, 3), the image to draw on bbox: tuple of (x_min, y_min, x_max, y_max) or list [x_min, y_min, x_max, y_max] color: tuple of (B, G, R), the color of the rectangle thickness: int, thickne

(image, bbox, color=(0, 255, 0), thickness=2)

# draw rectangle in an image

def draw_rectangle(image, bbox, color=(0, 255, 0), thickness=2):
    """
    Draw a rectangle on an image.
    
    Args:
        image: numpy array of shape (H, W, 3), the image to draw on
        bbox: tuple of (x_min, y_min, x_max, y_max) or list [x_min, y_min, x_max, y_max]
        color: tuple of (B, G, R), the color of the rectangle
        thickness: int, thickness of the rectangle lines
    
    Returns:
        numpy array: the image with rectangle drawn
    """
    x_min, y_min, x_max, y_max = [int(coord) for coord in bbox]
    cv2.rectangle(image, (x_min, y_min), (x_max, y_max), color, thickness)
    return image

def draw_rectangles(image, bboxes, colors=None, thickness=2):
    """