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

Question

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

тоже не помогло

Comments

[Модератор]

Фрагменты кода надо размещать в виде текста и оборачивать тэгом code для корректного отображения. Удобно делать кнопкой </>
Это обязательно, см.п.3.8 Регламента.
Сюда же относится traceback, ввод и вывод в консоли и другая структурированная текстовая инфа.

[Renat871]

не знал что traceback'и тоже надо в code оборачивать

Answer 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)

Comments

[Renat871]

Тот код которий Ви мне скинули возвращает следующее

Как можно обвести обьект квадратом и узнать точние координати центра обьекта?

[Vindicar]

Renat871, посмотреть и подумать. В массиве 3 строки по 6 значений с одинаковой структурой. Первые 4 значения выглядят как координаты (целые значения в пределах размера изображения). Первая пара наверняка левый верхний угол. Вторая пара слишком велика для ширины (угол выйдет за рамки изображения, оно у тебя 736х800), значит, это скорее всего правый нижний угол.
С учётом цикла for в приведённом коде, получаем массив размером 1х3х6. Т.е. код будет типа

for row in detections[0]:
    x1, y1, x2, y2 = map(int, row[:4])

Дальше сам. Как нарисовать прямоугольник на изображении по известным координатом - гуглится. Нужно будет уточнить в строке сначала идёт x, потом y, или наоборот.