iris_detection.py

# Questo lavoro è concesso in licenza MIT.
# Copyright (c) 2013-2023 OpenMV LLC. Tutti i diritti riservati.
# https://github.com/openmv/openmv/blob/master/LICENSE
#
# Esempio di rilevamento dell'iride 2
#
# Questo esempio mostra come trovare lo sguardo (rilevamento della pupilla) dopo aver trovato
# gli occhi in un'immagine. Questo script usa la funzione find_eyes che determina
# il punto centrale della roi che dovrebbe contenere una pupilla. Lo fa sostanzialmente
# trovando il centro dell'area più scura nella roi dell'occhio, che è il centro della pupilla.
#
# Nota: questo script non rileva prima un volto, usalo con l'obiettivo tele.

import csi
import time
import image

csi0 = csi.CSI()
csi0.reset()
csi0.contrast(3)
csi0.gainceiling(16)
csi0.framesize(csi.VGA)
# Binning/ritaglio dell'immagine a 200x100, che offre più dettagli con meno dati da elaborare
csi0.window((220, 190, 200, 100))
csi0.pixformat(csi.GRAYSCALE)

# Carica Haar Cascade
# Per impostazione predefinita verranno usati tutti gli stage; stage inferiori sono più veloci ma meno accurati.
eyes_cascade = image.HaarCascade("/rom/haarcascade_eye.cascade", stages=24)
print(eyes_cascade)

# Clock FPS
clock = time.clock()

while True:
    clock.tick()
    # Catturare lo snapshot
    img = csi0.snapshot()
    # Trova gli occhi!
    # Nota: un fattore di scala più basso riduce maggiormente l'immagine e rileva oggetti più piccoli.
    # Una soglia più alta comporta un tasso di rilevamento più elevato, con più falsi positivi.
    eyes = img.find_features(eyes_cascade, threshold=0.5, scale=1.5)

    # Trova l'iride
    for e in eyes:
        iris = img.find_eye(roi=e)
        img.draw_rectangle(e)
        img.draw_cross(iris)

    # Stampa gli FPS.
    # Nota: l'FPS effettivo è più alto; lo streaming del framebuffer lo rallenta.
    print(clock.fps())