dronefreak/gcnet-s-cityscapes

GCNet-S выполняет семантическую сегментацию дорожных RGB-изображений: каждому пикселю назначается один из 19 классов Cityscapes, на выходе получается карта разметки. Это вариант Golden Cudgel Network; издатель зеркала указывает примерно 9,2 млн параметров. Веса оригинальной модели сохранены в PTH/FP32; из тренировочного checkpoint убраны служебные состояния, а тензоры модели, по заявлению издателя, сохранены. Для локального запуска нужна реализация GCNet из встроенной в исходный репозиторий версии mmsegmentation вместе с PyTorch, MMCV и MMEngine: стандартный пакет mmsegmentation этой архитектуры не содержит. Пример источника поддерживает загрузку конфига и весов, inference одного кадра и сохранение цветной маски. Назначение — исследование сегментации дорожных сцен; скорость и качество на другом оборудовании не подтверждены. Веса зеркалирует dronefreak, исходный проект — GCNet.