Установка YOLOv11
Установка
pip install ultralytics
Виды доступных моделей для V11
- YOLOv11n - самая легкая и быстрая модель
- YOLOv11s
- YOLOv11m
- YOLOv11l
- YOLOv11x - тяжелая модель требующая больших вычислительных мощностей
Запуск программы для детектирования объектов по камере
from ultralytics import YOLO
import cv2
import numpy as np
# Загрузка модели
model = YOLO("yolo11n.pt")
camera = cv2.VideoCapture(0)
ret,frame = camera.read()
cv2.imshow('photo', frame)
camera.release()
results = model(frame, save=True)[0]
cv2.waitKey(0)
cv2.destroyAllWindows()
После запуска модель скачивается автоматически.
Результат детектирования лежит в runs/detect/predict
Запуск программы для самостоятельного анализа результатов распознавания
from ultralytics import YOLO
import cv2
import numpy as np
# Загрузка модели
model = YOLO("yolo11n.pt")
camera = cv2.VideoCapture(0)
ret,frame = camera.read()
cv2.imshow('photo', frame)
camera.release()
# Детекция объектов на изображении
results = model(frame, save=True)[0]
classes = results.boxes.cls.cpu().numpy()
pred = results.boxes.conf.cpu().numpy()
boxes = results.boxes.xyxy.cpu().numpy().astype(np.int32)
print(classes)
print(boxes)
print(pred)
cv2.waitKey(0)
cv2.destroyAllWindows()
Начать дискуссию