Проблема навчання моделі YOLO

Усі статті, обговорення, новини про AI — в одному місці. Підписуйтеся на DOU | AI!

При навчанні моделі з кожною наступною епохою значення mAP50 та mAP50-90 не міняються і дорівнюють 0 + з’являються значення nan Підкажіть що робити!!!

Epoch GPU_mem box_loss cls_loss dfl_loss Instances Size
49/100 2.37G nan nan nan 10 640: 100%|██████████| 9/9 [00:08<00:00, 1.10it/s]
Class Images Instances Box(P R mAP50 mAP50-95): 100%|██████████| 1/1 [00:00<00:00, 4.66it/s]
all 15 14 0 0 0 0

👍ПодобаєтьсяСподобалось0
До обраногоВ обраному0
LinkedIn
Дозволені теги: blockquote, a, pre, code, ul, ol, li, b, i, del.
Ctrl + Enter
Дозволені теги: blockquote, a, pre, code, ul, ol, li, b, i, del.
Ctrl + Enter

Який розмір train, validation сетів? Розмір зображень на вході?

А що на картинках?
Бо дуже мало як для діпльонінга.
Тренована модель чи з нуля?
В залежності від відповіді і стратегію треба обирати.
З нуля? — треба набагато більше зображень, як мінімум тисячі.
Тренована модель? — можна менше, але всеодно більше, під 1000 (хоча залежить від розміру зображень і кількості фіч)
Якщо можна отримати більше матеріалу — шукайте.
Якщо ні — препроцесінг, аугментація. Напишіть скрипт який повертає зображення, міняє масштаб, кольори, додає шум, і тд і додайте до датасету. Зробіть хоча б під 1к так. Можна погратись з кількістю епох.
А, ну і задача яка? Класифікація? Детекція?

Підписатись на коментарі