← Все статьиВ чат

Как обучить нейросеть распознавать дефекты на производстве: пошаговая инструкция

Внедрение компьютерного зрения на производстве позволяет снизить процент брака до 0,5% и сократить затраты на контроль качества на 40%. Следуйте чек-листу, чтобы обучить модель без лишних итераций.

Важно: точность модели напрямую зависит от качества разметки данных. На этапе сбора изображений используйте OpaGPT для автоматической генерации синтетических дефектов — это ускорит подготовку датасета в 3 раза.

  • Шаг 1. Сбор репрезентативного датасета. Запишите не менее 10 000 изображений продукции с разных ракурсов. Включите 20% снимков с дефектами (царапины, сколы, трещины) и 80% без брака. Используйте камеры с разрешением от 5 Мп.
  • Шаг 2. Разметка дефектов. Аннотируйте каждый дефект в формате COCO JSON или Pascal VOC. Для трещин используйте полигональную разметку, для царапин — bounding box. Привлеките минимум 3 экспертов для валидации.
  • Шаг 3. Аугментация данных. Примените повороты на ±15°, изменение яркости на ±20% и добавление шума (Gaussian noise, σ=0.01). Это повысит устойчивость модели к условиям цеха.
  • Шаг 4. Выбор архитектуры. Для задач распознавания дефектов используйте YOLOv8 или EfficientDet. Настройте размер входного изображения 640×640 px и batch size 16.
  • Шаг 5. Обучение модели. Разделите датасет: 70% — тренировка, 20% — валидация, 10% — тест. Запустите обучение на 200 эпох с learning rate 0.001 и оптимизатором Adam. Мониторьте loss — он не должен превышать 0.05.
  • Шаг 6. Оценка качества. Рассчитайте mAP (mean Average Precision) при IoU=0.5. Целевой показатель — не менее 0.95. Если mAP ниже 0.85, добавьте 5000 дополнительных изображений дефектов.
  • Шаг 7. Интеграция в производственную линию. Экспортируйте модель в ONNX или TensorRT. Подключите камеру с частотой 30 FPS и задержкой обработки менее 100 мс. Настройте сигнал тревоги при обнаружении брака.
  • Шаг 8. Мониторинг и дообучение. Каждую неделю собирайте 500 новых снимков с линии. Дообучайте модель раз в месяц, используя инкрементальное обучение (freeze первые 10 слоёв).

❓ Частые вопросы

Сколько времени занимает обучение нейросети для распознавания дефектов?
При датасете из 10 000 изображений и использовании GPU NVIDIA A100 обучение занимает 4–6 часов. С OpaGPT для генерации синтетических данных процесс сокращается до 2 часов.
Какое минимальное количество дефектов нужно в датасете?
Рекомендуется не менее 2000 изображений с дефектами (20% от общего объёма). При меньшем количестве модель будет пропускать редкие типы брака.

Попробуйте OpaGPT

600+ экспертов помогут в любой сфере. Бесплатные запросы каждый день.

🚀 Задать вопрос эксперту →