Классовый дисбаланс в сегментации изображений - PullRequest
0 голосов
/ 26 марта 2020

В настоящее время я работаю над проектом сегментации изображений. Изображения, над которыми я работаю, имеют разрешение 4k, а количество фоновых пикселей чрезвычайно велико по сравнению с пикселями переднего плана. Так как тренироваться на таких больших изображениях невозможно, я разбиваю изображение на случайные кадры 256x256. Когда я делаю случайную обрезку, я создаю равное количество изображений с объектами и только фоном, чтобы сбалансировать передний план и фон. Я использую потерю расстояния Jaccard. Мои кривые потерь при обучении и проверке выглядят очень хорошо Проверка и потери при обучении . Но, когда я прогнозирую изображения 4k для оценки, я вижу много ложных срабатываний. Любые предложения будут оценены.

...