gif_on_face_detection.py

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

import csi
import image
import time
import gif
import machine
import random

csi0 = csi.CSI()
csi0.reset()  # Инициализация датчика камеры.
csi0.pixformat(csi.GRAYSCALE)  # Установить формат пикселей RGB565 (или GRAYSCALE)
csi0.framesize(csi.QVGA)  # Установить размер кадра QVGA
csi0.snapshot(time=2000)  # Подождать, пока настройки вступят в силу.

led = machine.LED("LED_RED")

# Загрузить каскад Хаара (HaarCascade) для обнаружения лиц. Это объект, который OpenMV Cam
# может использовать для обнаружения лиц с помощью метода find_features() ниже. В OpenMV
# Cam уже встроен каскад Хаара frontalface. По умолчанию загружаются все этапы
# каскада Хаара. Однако вы можете уменьшить количество этапов, чтобы ускорить
# обработку за счёт точности. Каскад Хаара frontalface содержит 25
# этапов.
face_cascade = image.HaarCascade("/rom/haarcascade_frontalface.cascade", stages=25)

while True:
    print("About to start detecting faces...")
    csi0.snapshot(time=2000)  # Дать пользователю время подготовиться.

    print("Now detecting faces!")

    diff = 10  # Мы будем считать, что лицо обнаружено после 10 кадров.
    while diff:
        img = csi0.snapshot()
        # Порог может быть от 0.0 до 1.0. Более высокий порог даёт
        # более высокую частоту обнаружения, но с большим числом ложных срабатываний. Значение scale
        # управляет масштабом сопоставления, позволяя обнаруживать более мелкие лица.
        faces = img.find_features(face_cascade, threshold=0.5, scale=1.5)

        if faces:
            diff -= 1
            for r in faces:
                img.draw_rectangle(r)

    led.on()
    g = gif.Gif("example-%d.gif" % random.getrandbits(32), loop=True)

    clock = time.clock()  # Отслеживает FPS.
    for i in range(100):
        clock.tick()
        # clock.avg() возвращает миллисекунды между кадрами - задержка gif указывается в
        g.add_frame(csi0.snapshot(), delay=int(clock.avg() / 10))  # сотых долях секунды.
        print(clock.fps())

    g.close()
    led.off()
    print("Restarting...")