Задача «Разработка алгоритма определения железнодорожной колеи и подвижного состава для предотвращения чрезвычайных ситуаций на железной дороге»
В рамках чемпионата требуется создать алгоритм, определяющий элементы дорожной инфраструктуры: колею (рельсошпальную решетку) и подвижной состав (локомотивы, грузовые вагоны, пассажирские вагоны).
За основу была взята архитектура DeepLabV3Plus + предобученная модель xception71 из пакета segmentation_models_pytorch.
Применялось много аугментаций. В качестве функции потерь использовался dice-loss + bce лосс (в качестве целевого класса для пикселя используется наличие/отсутствие любой маски).
Обучение длилось несколько дней.