def draw_box(image, bbox, color, lw):
    """Draw RGB(A) `color` bounding box on image array."""
    y1, x1, y2, x2 = bbox
    image[y1 : y1 + lw, x1:x2] = color
    image[y2 : y2 + lw, x1:x2] = color
    image[y1:y2, x1 : x1 + lw] = color
    image[y1:y2, x2 : x2 + lw] = color
    return image