from ultralytics import YOLO
import cv2
# Загрузка предобученной модели YOLOv8n
model = YOLO('yolov8n.pt')
# Анализ изображения
results = model('image.jpg')
# Получение первого результата (так как обрабатываем одно изображение)
result = results[0]
# Извлечение координат и информации о найденных объектах
boxes = result.boxes # Объект Boxes, содержащий информацию о bounding boxes
# Отображение результатов на изображении
# Метод .plot() возвращает изображение в формате numpy array (BGR)
annotated_image = result.plot()
# Сохранение итогового изображения с помощью OpenCV
cv2.imwrite('result.jpg', annotated_image)
print("Обработка завершена. Результат сохранён в 'result.jpg'")