02-ago: [proyecto] vigilancia - alineacion facial ArcFace (landmarks YuNet) + consenso 3 frames en alerta + enroll sin rellenar nombres fantasma. Prueba 45s: INES reconocida 80% frames

vigilancia/enroll.py

@@ -35,9 +35,9 @@ def capturar_cuerpo(r, frame):
 
 
 def main():
-    nombres = sys.argv[1:3]
-    while len(nombres) < 2:
-        nombres.append(f"Hija{len(nombres) + 1}")
+    nombres = sys.argv[1:]
+    if not nombres:
+        nombres = ["Hija1", "Hija2"]
     print(f"Registrando: {nombres}")
 
     r = rec.cargar_reconocedor()
@@ -102,6 +102,9 @@ def main():
                 cv2.imwrite(os.path.join(carpeta, "cara_ref.jpg"),
                             frame[max(0, y - 20):y + h + 20, max(0, x - 20):x + w + 20])
                 print(f"  Cara de {nombre} registrada (score {caja[-1]:.2f})")
+                if os.path.exists(os.path.join(carpeta, "body.npy")):
+                    print(f"  {nombre} ya tenia cuerpo; registro completado.")
+                    break
                 fase = "cuerpo"
 
             elif fase == "cuerpo" and tecla == ord("b"):