YoloV8
This commit is contained in:
@@ -0,0 +1,37 @@
|
||||
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()
|
||||
Reference in New Issue
Block a user