Fire forest segmentation

Проект предназначен для оперативного реагирования пожарных бригад в условиях повышенной пожарной опасности. Алгоритм сегментирует возгорания на два класса: огонь и дым. Как только алгоритм определяет класс, дрон автоматически вызывает пожарную бригаду. Мое решение реализовано с использованием доработанной архитектуры нейронной сети UNet, которая была обучена в среде PyTorch. Средний loss на тестовой выборке 0.33 Обучал я на размеченном датасете с roboflow