Загрузка данных


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'")