Как вывести через print название обьекта, которое распознала модель ИИ?

Question

[Filmans]

Взял код из документации, вот код:

from ultralytics import YOLO
 
# Load a model
model = YOLO('best.pt')  # pretrained YOLOv8n model
 
# Run batched inference on a list of images
results = model(['ggg.jpg'])  # return a list of Results objects
 
# Process results list
for result in results:
    boxes = result.boxes  # Boxes object for bbox outputs
    masks = result.masks  # Masks object for segmentation masks outputs
    keypoints = result.keypoints  # Keypoints object for pose outputs
    probs = result.probs  # Class probabilities for classification outputs

Прога должна через print в терминале вывести. что увидела модель ИИ на картинке.

Пример: На изображении изображен: roses с вероятностью 1.00

Как это сделать. подскажите пж.

Comments

[Максим Припадчев]

Что такое 1 это один процент или 100 процентов?

[Filmans]

Максим Припадчев, Да, так пишет модель ИИ, когда распознала обьект.
1.00 - это 100 процентов вероятность

Answer 1

[Максим Припадчев]

Пример: На изображении изображен: roses с вероятностью 1.00

Если мы уверены в чем то на 100 процентов это не вопрос статистики, как следствие любая правильно натренированная модель не может выдавать "с вероятностью сто процентов". И это пожалуй самое интересное в твоем вопросе.

Ну а по кусочку кода от gpt и комментировать, что то тяжело. Если допустить что results содержит что то толковое, то потом просто происходит итерация с бессмысленным перезаписыванием переменных.

Comments

[Filmans]

Код не от gpt, а взят из документации ultralytics: https://docs.ultralytics.com/modes/predict/

но даже если нельзя поставить точную вероятность, то может можно как-то сделать, чтобы программа вывела, что за обьект изображен на картинке после выполнения обработки через мою модель ИИ.

[Максим Припадчев]

CODEFER, собери объекты внутри цикла в список или словарь. Если эта модель что то делает, то вероятности содержаться в probs. и выведи их на печать, это совсем основы.

[Filmans]

Максим Припадчев, Собрал обьекты в словарь. Теперь надо построить логику, чтобы принт вывел не просто весь словарь, а только название обьекта, которое распознала модель ИИ.

[Максим Припадчев]

CODEFER, словарь это структура ключ:значение. Что у тебя является ключами, а что значениями?

[Filmans]

Максим Припадчев, Моя модель ИИ распознает некоторые виды цветов.

Поэтому словарь выглядит так:

flowers = {
    "daisy": "ромашка",
    "dandelion": "одуванчик",
    "roses": "розы",
    "sunflowers": "подсолнухи",
    "tulips": "тюльпаны"
}

[Максим Припадчев]

CODEFER, Ну что то не то у тебя собралось. У тебя должны были собраться ключи названия, а значения вероятности. А у тебя ключ = название на английском, значения = названия на русском. Ты посмотрел что в probs?

[Filmans]

Максим Припадчев, Вероятность пока не трогаем.

Сначала бы вывести название цветка, которое распознала модель.

А ключи которые на английском, это те самые названия цветов, на английском их модель принимает.

[Максим Припадчев]

CODEFER, Почитал, я твою так называемую документацию, все равно считаю что кусочки кода из этой документации писал gpt. Люди такого не пишут, да и API идиотское. попробуй results = model.predict(source="0")

[Максим Припадчев]

CODEFER, Все нашел я все что тебе нужно, https://www.freecodecamp.org/news/how-to-detect-ob... . Вот там и объекты и вероятнотси, в общем медленно пошагово, как все получить из того что умеет модель, даже если английский не знаешь, то переводи и вчитывайся кусочки маленькие и очень простые. А "документация" у меня большие сомнения, что она создана человеком - это что то странное.