38 lines
1.2 KiB
Python
38 lines
1.2 KiB
Python
import cv2
|
|
from ultralytics import YOLO
|
|
|
|
# Загрузим YOLOv8 модель (если ты скачал .pt файл вручную, укажи путь)
|
|
model = YOLO('yolov8n.pt') # автоматически скачается, если файла нет
|
|
|
|
# Открываем веб-камеру
|
|
cap = cv2.VideoCapture(0)
|
|
|
|
while True:
|
|
ret, frame = cap.read()
|
|
if not ret:
|
|
break
|
|
|
|
# Детекция объектов
|
|
results = model(frame)[0]
|
|
|
|
for result in results.boxes:
|
|
cls_id = int(result.cls[0])
|
|
conf = float(result.conf[0])
|
|
if model.names[cls_id] == 'person':
|
|
x1, y1, x2, y2 = map(int, result.xyxy[0])
|
|
# Зелёная рамка
|
|
cv2.rectangle(frame, (x1, y1), (x2, y2), (0, 255, 0), 2)
|
|
# Текст
|
|
label = f'{model.names[cls_id]} {conf:.2f}'
|
|
cv2.putText(frame, label, (x1, y1 - 5), cv2.FONT_HERSHEY_SIMPLEX, 0.5, (0,255,0), 1)
|
|
|
|
# Показываем результат
|
|
cv2.imshow("Detection", frame)
|
|
|
|
# Выход — по клавише ESC
|
|
if cv2.waitKey(1) & 0xFF == 27:
|
|
break
|
|
|
|
cap.release()
|
|
cv2.destroyAllWindows()
|