face_eye_detection.py

# Este trabalho está licenciado sob a licença MIT.
# Copyright (c) 2013-2023 OpenMV LLC. Todos os direitos reservados.
# https://github.com/openmv/openmv/blob/master/LICENSE
#
# Exemplo de Detecção de Rosto e Olhos
#
# Este script usa o detector frontalface embutido para encontrar um rosto e depois
# os olhos dentro do rosto. Se você quiser determinar a direção do olhar, veja o
# script iris_detection para um exemplo de como fazer isso.

import csi
import time
import image

csi0 = csi.CSI()
csi0.reset()
csi0.contrast(1)
csi0.gainceiling(16)
csi0.framesize(csi.HQVGA)
csi0.pixformat(csi.GRAYSCALE)

# Carrega o Haar Cascade
# Por padrão, isso usará todos os estágios; menos estágios é mais rápido, mas menos preciso.
face_cascade = image.HaarCascade("/rom/haarcascade_frontalface.cascade", stages=25)
eyes_cascade = image.HaarCascade("/rom/haarcascade_eye.cascade", stages=24)
print(face_cascade, eyes_cascade)

# Relógio de FPS
clock = time.clock()

while True:
    clock.tick()

    # Captura o snapshot
    img = csi0.snapshot()

    # Encontre um rosto!
    # Nota: Um fator de escala menor reduz mais a imagem e detecta objetos menores.
    # Um limiar mais alto resulta em uma taxa de detecção maior, com mais falsos positivos.
    objects = img.find_features(face_cascade, threshold=0.5, scale=1.5)

    # Desenha os rostos
    for face in objects:
        img.draw_rectangle(face)
        # Agora encontra os olhos dentro de cada rosto.
        # Nota: Use um limiar mais alto aqui (mais detecções) e escala menor (para encontrar objetos pequenos)
        eyes = img.find_features(
            eyes_cascade, threshold=0.5, scale=1.2, roi=face
        )
        for e in eyes:
            img.draw_rectangle(e)

    # Imprime o FPS.
    # Nota: O FPS real é maior, transmitir o FB torna mais lento.
    print(clock.fps())