You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
37 lines
1.2 KiB
37 lines
1.2 KiB
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()
|
|
|