Стажировка по проекту
Создание системы распознавания взрывоопасных предметов для перегружателей металлического лома
Компания: ООО «ПМХ-Втормет»
Суть проекта
Задача
Для чего
Вводные данные
Создание надежной и точной системы компьютерного зрения для выявления визуально опасных объектов на перегружателе лома, обеспечивающей безопасность и удобство работы оператора
- Датасет, состоящий из 2000 изображений визуально опасных объектов, снятых на мобильные устройства
- Используемые технологии: YOLO v8, Label Studio, PyQt6, OpenCV
- Обученная и протестированная модель YOLO для детекции ВОП
- Разработанное десктопное приложение на PyQt6 с возможностью мониторинга и обработки видеопотоков в реальном времени.
- Полностью размеченный и подготовленный к обучению и тестированию датасет
Выходные данные
Снижение риска пропуска визуально опасных объектов, улучшение безопасности производственного процесса, а также повышение точности и скорости работы оператора
Проект «Втормет» продемонстрировал успешное взаимодействие команды, высокую вовлеченность стажёров и их готовность решать сложные задачи
Особо выделяются заслуги Олега Шамаева, Алексея Герасимова и Зорислава Выймова, чьи усилия были решающими для успешной реализации проекта. Прогресс проекта создаёт прочную основу для дальнейшего развития и совершенствования системы, позволяя интегрировать её в реальные производственные условия
Особо выделяются заслуги Олега Шамаева, Алексея Герасимова и Зорислава Выймова, чьи усилия были решающими для успешной реализации проекта. Прогресс проекта создаёт прочную основу для дальнейшего развития и совершенствования системы, позволяя интегрировать её в реальные производственные условия
результат
Участники
1 команды проекта
1 команды проекта
- Терещенко алексейТимлид проекта
- коваленко сергейПомощник тимлида
- шамаев олегЗанимается внедрением AI в электротехнической компании
- Разметка датасета с использованием инструмента Label Studio
- Проведение экспериментов по обучению модели YOLOv8-OBB (Oriented Bounding Box)
- Разработка десктопного приложения на фреймворке PyQt6:
- трансляция четырех видеопотоков,
- вкладка для настройки параметров,
- выбор области для детекции и исключение определенных зон из процесса детекции,
- запись результатов детекции в базу данных или файл
- герасимов алексей
- Взял на себя всю работу по созданию и обучению моделей YOLO для задачи сегментации ВОП
- Провёл серию экспериментов с обучением моделей, разработал и внедрил оптимизированные методы предобработки и аугментации данных, что значительно повысило точность детекции и общую производительность системы
- Экспертиза в области машинного обучения, в частности, глубоких нейронных сетей и методов компьютерного зрения, а также умение решать сложные технические задачи стали ключевыми факторами успешной реализации проекта
- выймов зориславБинес-аналитик
- Провёл анализ и сортировку датасета по предполагаемым классам объектов
- Установил и освоил Label Studio для разметки данных
- Выполнил ручную разметку 1/3 датасета объектов заданных классов (ОВВ)
- Разработал Python-скрипт для конвертации разметки из JSON в формат для обучения YOLOv8 и автоматической подготовки данных
- Обучил модель YOLO для детекции опасных предметов, предназначенную для авторазметки оставшейся части датасета
- Автоматически разметил 2/3 датасета с помощью обученной модели и провёл проверку и коррекцию
- Создал Python-скрипт для объединения данных в коллажи для улучшения условий обучения модели на удалённых объектах
- любарский сергейПредприниматель, CEO, CTO
- Изучил техническое задание и разработал собственный вариант архитектуры решения
- Разметил 200 снимков Взрывоопасных предметов (ВОП) — часть набора данных для обучения нейросети
- Разработал и протестировал модуль на Python с использованием PyQt6 для сегментации ВОП на базе модели YOLOv8
- Предложил несколько альтернативных методов приведения набора данных в соответствие с требованиями технического задания заказчика
- Участвовал в командной работе над проектом
- валеев рамильМенеджер, направление нефтехимия
- Разметка датасета заказчика:
• Автоматическая разметка с использованием нейросети, обученной на ручной разметке
2. Разработка функций для работы с датасетом:
• Визуальная проверка качества разметки, слияние данных, разделение на наборы, автоматическое создание YAML файла для YOLO
• Подсчёт и визуализация сбалансированности классов для задач Object Detection
• Функция реализующая метод аугментации «мозаикой» с возможностью выбора произвольного размера выходного файла и настраиваемыми параметрами коллажа
3. Тестовые обучения сетей YOLO OBB - вавилов антонSales Engineer в компании SMC
- Собрал и разметил датасет в Label Studio
- Обучил модель Yolov8
- Подготовил итоговую презентацию