Для обучения модели команда проекта собирала видеоданные с реальных производственных процессов предприятия. В датасет вошли различные сценарии:
- сотрудники в касках и без касок;
- использование сигнальных жилетов;
- работа персонала в опасных зонах;
- перемещение техники и людей;
- различные условия освещения и загруженности площадки.
После сбора данных специалисты обучили модели компьютерного зрения распознавать сотрудников, элементы СИЗ и производственные зоны. Для повышения точности использовались методы аугментации данных и дополнительная разметка кадров.
В результате была обучена нейросеть для промышленной безопасности, способная автоматически фиксировать нарушения требований охраны труда в реальном времени.