02-ago: [proyecto] Sistema vigilancia Tapo con deteccion movimiento/persona + notificacion Telegram/Alexa

vigilancia/detector_persona.py

@@ -0,0 +1,15 @@
+from ultralytics import YOLO
+
+
+class DetectorPersona:
+    def __init__(self, modelo="yolov8n.pt", confianza=0.5):
+        self.model = YOLO(modelo)
+        self.confianza = confianza
+
+    def hay_persona(self, frame) -> bool:
+        resultados = self.model.predict(frame, verbose=False, conf=self.confianza)
+        for r in resultados:
+            for c in r.boxes.cls:
+                if int(c) == 0:
+                    return True
+        return False