Чому використовують anchor boxes в задачах детекції об'єктів

Anchor boxes — це фундаментальна концепція в сучасних моделях детекції об'єктів, таких як Faster R-CNN та YOLO. Вони слугують заздалегідь визначеними еталонними рамками, які допомагають виявляти об'єкти різних розмірів і співвідношень сторін, що робить процес детекції швидшим і надійнішим.

Замість того, щоб знаходити об'єкти з нуля, моделі використовують anchor boxes як початкові точки, коригуючи їх для кращого співпадіння з виявленими об'єктами. Такий підхід підвищує ефективність і точність, особливо при роботі з об'єктами різних масштабів.

Різниця між anchor box та bounding box

• Anchor box: заздалегідь визначений шаблон, який використовується як еталон під час детекції об'єктів;

• Bounding box: фінальна передбачена рамка після коригування anchor box для відповідності реальному об'єкту.

На відміну від bounding boxes, які динамічно коригуються під час передбачення, anchor boxes фіксуються у певних позиціях до початку детекції об'єктів. Моделі навчаються уточнювати anchor boxes, змінюючи їх розмір, положення та співвідношення сторін, у підсумку перетворюючи їх на фінальні bounding boxes, які точно відображають знайдені об'єкти.

Як мережа генерує anchor boxes

Anchor boxes застосовуються не безпосередньо до зображення, а до карт ознак, отриманих із зображення. Після виділення ознак набір anchor boxes розміщується на цих картах ознак, змінюючись за розміром і співвідношенням сторін. Вибір форм anchor boxes є критичним і передбачає баланс між виявленням малих і великих об'єктів.

Для визначення розмірів anchor boxes моделі зазвичай використовують поєднання ручного вибору та алгоритмів кластеризації, таких як K-Means, для аналізу набору даних і визначення найпоширеніших форм і розмірів об'єктів. Ці заздалегідь визначені anchor boxes потім застосовуються в різних місцях на картах ознак. Наприклад, модель для виявлення об'єктів може використовувати anchor boxes розмірів (16x16), (32x32), (64x64), зі співвідношеннями сторін 1:1, 1:2, and 2:1.

Після визначення anchor boxes вони застосовуються до карт ознак, а не до оригінального зображення. Модель призначає кілька anchor boxes кожному місцю на карті ознак, охоплюючи різні форми та розміри. Під час навчання мережа коригує anchor boxes, прогнозуючи зсуви, уточнюючи їх розмір і положення для кращого відображення об'єктів.

Від anchor box до bounding box

Після призначення anchor boxes об'єктам модель прогнозує зсуви для їх уточнення. Ці зсуви включають:

• Коригування координат центру box;

• Масштабування ширини та висоти;

• Зміщення box для кращого вирівнювання з об'єктом.

Застосовуючи ці перетворення, модель перетворює anchor boxes на фінальні bounding boxes, які максимально точно відповідають об'єктам на зображенні.

Підходи без використання або зі зменшеною кількістю anchor-боксів

Хоча anchor-бокси широко застосовуються, деякі моделі прагнуть зменшити залежність від них або повністю їх усунути:

• Anchor-free методи: моделі, такі як CenterNet та FCOS, визначають розташування об'єктів без попередньо заданих anchor-боксів, що зменшує складність;

• Підходи зі зменшеною кількістю anchor-боксів: EfficientDet та YOLOv4 оптимізують кількість anchor-боксів, забезпечуючи баланс між швидкістю виявлення та точністю.

Ці підходи спрямовані на підвищення ефективності виявлення об'єктів при збереженні високої продуктивності, особливо для задач реального часу.

Підсумовуючи, anchor-бокси є важливою складовою виявлення об'єктів, допомагаючи моделям ефективно знаходити об'єкти різних розмірів і співвідношень сторін. Проте нові розробки досліджують способи зменшення або усунення anchor-боксів для ще швидшого та гнучкішого виявлення.