Чтобы использовать YOLO («Вы смотрите только один раз») в PyTorch Hub, вы можете выполнить следующие действия:

1. Установите PyTorch и torchvision. Убедитесь, что в вашей системе установлены PyTorch и torchvision. Вы можете установить их с помощью pip:

   pip install torch torchvision

2. Импортируйте необходимые библиотеки: в свой скрипт Python импортируйте необходимые библиотеки:

   import torch
   import torchvision
   # Load the YOLO model from PyTorch Hub
   model = torch.hub.load('ultralytics/yolov5', 'yolov5s')

3. Загрузить изображение. Вам необходимо загрузить изображение, на котором вы хотите выполнить обнаружение объектов. Убедитесь, что изображение имеет формат, поддерживаемый PyTorch (например, изображение PIL):

   from PIL import Image
   # Load the image
   image = Image.open('path/to/your/image.jpg')

4. Запустить обнаружение объектов: используйте загруженную модель YOLO для обнаружения объектов на изображении:

   # Perform object detection
   results = model(image)

   Переменная resultsбудет содержать обнаруженные объекты вместе с их ограничивающими рамками и показателями достоверности.

5. Обработка результатов. Вы можете обработать результаты для извлечения обнаруженных объектов и их свойств:

   # Extract bounding boxes, labels, and scores
   boxes = results.xyxy[0][:, :4]  # Bounding boxes
   labels = results.xyxy[0][:, 5]  # Labels (class indices)
   scores = results.xyxy[0][:, 4]  # Confidence scores

6. Отображение или использование результатов. Вы можете отображать результаты или использовать их для дальнейшего анализа или задач.