Skip to content

pixano.datasets.utils.mosaic

add_label_above(image, text, label_height=30)

Add a label at top left of image.

Source code in pixano/datasets/utils/mosaic.py 
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
def add_label_above(image, text, label_height=30):
    """Add a label at top left of image."""
    width, height = image.size

    label_img = Image.new("RGB", (width, label_height), (0, 0, 0))  # Fond noir
    draw = ImageDraw.Draw(label_img)

    # get Arial font, else system font
    try:
        font = ImageFont.truetype("arial.ttf", 20)
    except IOError:
        font = ImageFont.load_default()

    # Center text
    text_size = draw.textbbox((0, 0), text, font=font)  # Mesurer le texte
    text_width = text_size[2] - text_size[0]
    text_height = text_size[3] - text_size[1]

    text_x = (width - text_width) // 2
    text_y = (label_height - text_height) // 2

    # Draw text (white)
    draw.text((text_x, text_y), text, fill=(255, 255, 255), font=font)

    # Concat label and original image
    new_img = Image.new("RGB", (width, height + label_height))
    new_img.paste(label_img, (0, 0))  # Add label over
    new_img.paste(image, (0, label_height))  # put image under

    return new_img

arrange_grid(num_images)

Get optimal grid distribution (rows, cols) for num_images.

Source code in pixano/datasets/utils/mosaic.py 
26
27
28
29
30
def arrange_grid(num_images):
    """Get optimal grid distribution (rows, cols) for num_images."""
    cols = math.ceil(math.sqrt(num_images))
    rows = math.ceil(num_images / cols)
    return rows, cols

compute_average_size(images)

Compute images average size.

Source code in pixano/datasets/utils/mosaic.py 
33
34
35
36
37
38
def compute_average_size(images):
    """Compute images average size."""
    widths, heights = zip(*(img.size for img in images))
    avg_width = sum(widths) // len(images)
    avg_height = sum(heights) // len(images)
    return avg_width, avg_height

create_mosaic(source_dir, split, image_files, output_path, label_prefix='image', padding=10, label_height=30)

Create a mosaic from image_files with black background.

Source code in pixano/datasets/utils/mosaic.py 
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
def create_mosaic(
    source_dir: Path, split, image_files, output_path, label_prefix="image", padding=10, label_height=30
):
    """Create a mosaic from image_files with black background."""
    if not image_files:
        return

    need_split = False
    images = []
    for f in image_files:
        try:
            image = Image.open(source_dir / f)
        except FileNotFoundError:
            image = Image.open(source_dir / split / f)
            need_split = True
        image = image.convert("RGBA")
        images.append(image)

    rows, cols = arrange_grid(len(images))
    avg_width, avg_height = compute_average_size(images)

    images_resized = []
    for idx, img in enumerate(images):
        img_resized = img.resize((avg_width, avg_height), Image.LANCZOS)
        img_rgb = Image.new("RGB", img_resized.size, (0, 0, 0))
        img_rgb.paste(img_resized, mask=img_resized.split()[3])  # Apply alpha channel
        # Add label
        label_text = f"{label_prefix} {idx + 1}"
        img_labeled = add_label_above(img_rgb, label_text, label_height)
        images_resized.append(img_labeled)

    mosaic_width = cols * (avg_width + padding) - padding
    mosaic_height = rows * (avg_height + label_height + padding) - padding
    mosaic = Image.new("RGB", (mosaic_width, mosaic_height), (0, 0, 0))

    for idx, img in enumerate(images_resized):
        x_offset = (idx % cols) * (avg_width + padding)
        y_offset = (idx // cols) * (avg_height + label_height + padding)
        mosaic.paste(img, (x_offset, y_offset))

    if need_split:
        mosaic.save(source_dir / split / output_path, format="JPEG", quality=85, optimize=True)
    else:
        mosaic.save(source_dir / output_path, format="JPEG", quality=85, optimize=True)

generate_mosaic_name(image_files, extension='.jpg')

Generate a name for mosaic file whime keeping most common path / filename as possible.

Source code in pixano/datasets/utils/mosaic.py 
15
16
17
18
19
20
21
22
23
def generate_mosaic_name(image_files, extension=".jpg"):
    """Generate a name for mosaic file whime keeping most common path / filename as possible."""
    if not image_files:
        return "mosaic" + extension
    common_path = os.path.commonpath(image_files)
    filenames = [os.path.basename(f) for f in image_files]
    common_prefix = os.path.commonprefix(filenames).rstrip("_- ")
    mosaic_name = f"{common_prefix}_mosaic{extension}" if common_prefix else f"mosaic{extension}"
    return os.path.join(common_path, mosaic_name)

mosaic(source_dir, split, image_files, view_name)

Create a mosaic from input images.

Source code in pixano/datasets/utils/mosaic.py 
119
120
121
122
123
def mosaic(source_dir: Path, split: str, image_files: list[str], view_name: str) -> str:
    """Create a mosaic from input images."""
    mosaic_filename = generate_mosaic_name(image_files)
    create_mosaic(source_dir, split, image_files, mosaic_filename, label_prefix=view_name, label_height=30, padding=5)
    return mosaic_filename