image_drawing_with_custom_palette.py

# Эта работа лицензирована по лицензии MIT.
# Copyright (c) 2013-2023 OpenMV LLC. All rights reserved.
# https://github.com/openmv/openmv/blob/master/LICENSE
#
# Пример рисования изображения с пользовательской цветовой палитрой
#
# Этот пример показывает, как рисовать изображения в буфере кадра с пользовательской сгенерированной цветовой палитрой.

import csi
import image
import time

csi0 = csi.CSI()
csi0.reset()
csi0.pixformat(csi.GRAYSCALE)  # или GRAYSCALE...
csi0.framesize(csi.QQVGA)  # или QQVGA...
csi0.snapshot(time=2000)

clock = time.clock()

# цветовая палитра на самом деле является изображением, это позволяет использовать операции над изображением для создания палитр
# изображение должно иметь 256 записей, т.е. 256x1, 64x4, 16x16, и иметь формат rgb565

# Инициализировать исходные цвета палитры в изображении
palette_source_colors = [(255, 0, 0), (0, 255, 0), (0, 0, 255), (255, 0, 255)]
palette_source_color_image = image.Image(len(palette_source_colors), 1, csi.RGB565)
for i, color in enumerate(palette_source_colors):
    palette_source_color_image[i] = color

# Масштабировать изображение до ширины палитры и сгладить их
palette = image.Image(256, 1, csi.RGB565)
palette.draw_image(
    palette_source_color_image,
    0,
    0,
    x_scale=palette.width() / palette_source_color_image.width(),
)
palette.mean(int(palette.width() / palette_source_color_image.width() / 2))

while True:
    clock.tick()

    img = csi0.snapshot()
    # Получить копию изображения в оттенках серого перед преобразованием в цвет
    img_copy = img.copy()

    img.to_rgb565()

    palette_boundary_inset = int(csi0.width() / 40)
    palette_scale_x = (csi0.width() - palette_boundary_inset * 2) / palette.width()

    img.draw_image(img_copy, 0, 0, color_palette=palette)
    img.draw_image(
        palette,
        palette_boundary_inset,
        palette_boundary_inset,
        x_scale=palette_scale_x,
        y_scale=8,
    )
    img.draw_rectangle(
        (palette_boundary_inset, palette_boundary_inset, int(palette.width() * palette_scale_x), 8),
        color=(255, 255, 255),
        thickness=1,
    )

    print(clock.fps())