bike_rec.py

import cv2 
from ultralytics import YOLO
import winsound
import threading

video = cv2.VideoCapture(0)
modelo = YOLO('yolov8n.pt')
area = [390, 120, 600, 320]

alarme_ctl = False

def alarme():
    global alarme_ctl
    for _ in range(7):
        winsound.Beep(2500,500)
        

    alarme_ctl = False



while True:
    check, img = video.read()
    img2 = img.copy()
    cv2.rectangle(img2, (area[0], area[1]), (area[2], area[3]), (0,255,0), -1)
    resultado = modelo(img)

    for objetos in resultado:
        obj = objetos.boxes
        for dados in obj:
            x,y,w,h = dados.xyxy[0]
            x,y,w,h = int(x), int(y), int(w), int(h)
            cls = int(dados.cls[0])
            # centro da área de interesse:
            cx, cy = (x+w)//2, (y+h)//2 
            if cls == 1:
                cv2.rectangle(img, (x, y), (w, h), (255,0,0), 5)
                if cx>=area[0] and cx<=area[2] and cy>=area[1] and cy<=area[3]:
                    cv2.rectangle(img2, (area[0], area[1]), (area[2], area[3]), (0,0,255), -1)
                    cv2.rectangle(img, (70,20), (400, 60), (0,0,255), -1)
                    cv2.putText(img, "Invasor Detectado", (100,50), cv2.FONT_HERSHEY_SIMPLEX, 1, (255,255,255), 3)       
                    if not alarme_ctl:
                        alarme_ctl = True
                        threading.Thread(target=alarme).start()


             
 
    imgFinal = cv2.addWeighted(img2, 0.5, img, 0.5, 0)

    cv2.imshow('img', imgFinal)
    key = cv2.waitKey(1)
    if key == 27:
        break

video.release()
cv2.destroyAllWindows()