Интеллектуальная система анализа рентген-изображений транспорта для задач безопасности

Цель проекта

Целью проекта стала разработка нейросетевой системы сегментации и классификации транспортных средств, предназначенной для помощи операторам при досмотре автомобилей, проходящих через рентген-сканеры.

Описание

Команда «Наносемантики», включающая специалистов по разметке данных, инженеров CV и разработчиков нейросетевых решений, реализовала программное обеспечение на базе технологий компьютерного зрения и глубокого обучения. Система анализирует рентген-изображения транспорта и автоматически определяет класс транспортного средства, выделяет его составные части с указанием координат для последующей визуализации, а также выявляет дефекты и классифицирует их по типам. Результат работы формируется в формате JSON, который может быть интегрирован в существующие системы заказчика.

Для обучения моделей использовался датасет из 15 000 рентген-снимков, при этом около 9 000 изображений приходилось на грузовой транспорт. Такой дисбаланс был согласован с заказчиком, поскольку разметка грузовых автомобилей значительно сложнее из-за большего количества элементов. Все изображения предоставлялись в двух ракурсах ‒ вид сбоку и сверху, что также учитывалось при обучении моделей.

В процессе разработки использовались инструменты разметки CVAT и язык программирования Python, а также широкий набор технологий компьютерного зрения, включая методы классического CV, object detection, semantic и instance segmentation. Система адаптирована для работы на операционной системе Alt Рабочая Станция 10. Разработка пользовательского интерфейса в рамках проекта не предусматривалась, так как решение интегрируется в существующий контур заказчика.

Результаты

В ходе реализации проекта были достигнуты и превышены все ключевые метрики качества, заложенные в техническом задании.

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

В настоящее время «Наносемантика» реализует третий этапа проекта, который включает повышение качества сегментации частей транспорта и разработку облегченной версии решения на языке C++ для повышения производительности.

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