Почему YOLO добавляет «фоновый» класс к обнаружению объектов, даже если мы не добавили этот класс? Как его удалить?

Question

Почему YOLO добавляет «фоновый» класс к обнаружению объектов, даже если мы не добавили этот класс? Как его удалить?

Я пытаюсь обучить модель YOLO v8x на пользовательских данных. В матрице путаницы я увидел, что был добавлен фоновый класс, который я не аннотировал. Позже я заметил, что это происходит и с YOLO v5. Если у кого-нибудь есть идеи по этому поводу, не могли бы вы мне рассказать? И как удалить этот класс? а где можно проверить какие все классы учитываются в модели для обучения?

Answer 1

Вы можете получить доступ к списку классов модели, позвонивmodel.names. Он не будет содержать класс «фон», поскольку это концепция по умолчанию, определяющая ложноположительные прогнозы вашей модели, когда она видит несуществующие объекты в фоновом режиме. Не нужно пытаться его удалить. Полезно показать это в матрице путаницы, чтобы вы могли оценить производительность модели относительно того, имеет ли она тенденцию генерировать ложноположительные прогнозы и какие из ваших классов подвержены этой ошибке.