@Renat871

Как отобразить квадратом выделение нейронной сети?

Я обучил модель .pt. Я не знаю, как импортировать. Пробовал через Torch.hub.load - выбивает ошибку
model = torch.hub.load('ultralytics/yolov5', 'custom', './best.pt', force_reload=True)


Скриншот фрагмента кода удалён модератором.

Пробовал Через библиотеку YOLO
Что-то типа
model = yolo.model("best.pt")
model."забыл функцию"("foto.png")

тоже не помогло
  • Вопрос задан
  • 92 просмотра
Решения вопроса 1
@romazhan
Для отображения квадратом выделения нейронной сети в YOLO, вам нужно использовать функцию `predict`, которая принимает путь к изображению и возвращает список обнаруженных объектов с их координатами:
from yolo import YOLO

yolo = YOLO()

image_path = "foto.png"

detections = yolo.predict(image_path)

for detection in detections:
    print(detection)
Ответ написан
Пригласить эксперта
Ваш ответ на вопрос

Войдите, чтобы написать ответ

Войти через центр авторизации
Похожие вопросы