iris_detection.py

# Esta obra está licenciada bajo la licencia MIT.
# Copyright (c) 2013-2023 OpenMV LLC. Todos los derechos reservados.
# https://github.com/openmv/openmv/blob/master/LICENSE
#
# Ejemplo 2 de detección de iris
#
# Este ejemplo muestra cómo encontrar la dirección de la mirada (detección de pupila) después de encontrar
# los ojos en una imagen. Este script usa la función find_eyes, que determina
# el punto central de la roi que debería contener una pupila. Lo hace básicamente
# encontrando el centro del área más oscura en la roi del ojo, que es el centro de la pupila.
#
# Nota: Este script no detecta un rostro primero; úsalo con la lente telefoto.

import csi
import time
import image

csi0 = csi.CSI()
csi0.reset()
csi0.contrast(3)
csi0.gainceiling(16)
csi0.framesize(csi.VGA)
# Bin/recorta la imagen a 200x100, lo que da más detalle con menos datos que procesar
csi0.window((220, 190, 200, 100))
csi0.pixformat(csi.GRAYSCALE)

# Cargar cascada Haar
# Por defecto se usarán todas las etapas; menos etapas es más rápido pero menos preciso.
eyes_cascade = image.HaarCascade("/rom/haarcascade_eye.cascade", stages=24)
print(eyes_cascade)

# Reloj de FPS
clock = time.clock()

while True:
    clock.tick()
    # Captura una instantánea
    img = csi0.snapshot()
    # ¡Encontrar ojos!
    # Nota: Un factor de escala más bajo reduce más la imagen y detecta objetos más pequeños.
    # Un umbral más alto da como resultado una tasa de detección mayor, con más falsos positivos.
    eyes = img.find_features(eyes_cascade, threshold=0.5, scale=1.5)

    # Encontrar iris
    for e in eyes:
        iris = img.find_eye(roi=e)
        img.draw_rectangle(e)
        img.draw_cross(iris)

    # Imprime los FPS.
    # Nota: La velocidad de fotogramas real es mayor; transmitir el framebuffer lo hace más lento.
    print(clock.fps())