Челябинские студенты подготовили уникальный датасет для беспилотников  Оригинал

Команда студентов ЮУрГУ завершила работу над датасетом для автономного транспорта. В него вошли 10 тысяч размеченных изображений дорог, снятых преимущественно в Челябинске и Кургане.

Проектом руководил студент Роман Алиев, научным куратором стала доцент кафедры системного программирования Ольга Иванова. Разработчики хотели решить проблему, о которой часто говорят в теме беспилотников: зарубежные базы плохо учитывают российскую дорожную среду.

Материал собирали с помощью RGB-камеры, закрепленной на лобовом стекле автомобиля. Получилось более 300 видеофайлов общей длительностью свыше 1000 минут. После обработки в финальную выборку попали 10 тысяч изображений.

Разметка подробная: объекты выделяли попиксельно. Один участок дороги мог относиться сразу к нескольким классам — например, влажная грязь с листьями одновременно попадала в категории грязи и мусора.

Всего в датасете 14 классов без учета фона. Там есть ямы, трещины, колейность, заплатки, стыки асфальта, люки, трамвайные рельсы, разметка, лужи, мусор и другие элементы покрытия.

Над разметкой сначала вручную работали 19 студентов. Затем обучалась модель ИИ, которая заполняла аннотации на новых кадрах, а люди проверяли результат. В выборку специально добавили ночные кадры, дождь, плохой свет и влажную дорогу.

Сейчас часть данных — 20% — могут получить исследователи по запросу и согласованию с университетом.