def get_dominant_color()

in src/co_op_translator/utils/vision/image_utils.py [0:0]

• 16 lines of code
• 2 McCabe index (conditional complexity)

def get_dominant_color(image, bounding_box, palette_size=16):
    """
    Get the dominant (colour occurring with the highest pixel frequency) of a bounding box area in the image.

    Args:
        image (PIL.Image.Image): The image object.
        bounding_box (list): The bounding box coordinates.
        palette_size (int): The size of the color palette to reduce the image to.

    Returns:
        tuple: The dominant color (R, G, B).
    """
    cropped_image = image.crop(
        (
            min(bounding_box[::2]),
            min(bounding_box[1::2]),
            max(bounding_box[::2]),
            max(bounding_box[1::2]),
        )
    )

    # Crop, resize and use the palette to identify the dominant colour
    cropped_image.thumbnail((400, 400))
    paletted = cropped_image.convert("P", palette=Image.ADAPTIVE, colors=palette_size)
    palette = paletted.getpalette()
    color_counts = sorted(paletted.getcolors(), reverse=True)
    palette_index = color_counts[0][1]
    dominant_color = palette[palette_index * 3 : palette_index * 3 + 3]

    return tuple(dominant_color), cropped_image