Post #49
40



Как научили нейросеть отличать брак от сдвигов и стыков
Обещали рассказать, как научили нейросеть находить дефекты ленты сверхпроводника и не путать их со сдвигами, стыками и другими допустимыми артефактами производства.
👉 Начнем со сдвигов.
Во время перемотки лента может смещаться в кадре. Для оператора это не проблема, а вот алгоритм легко может принять такой сдвиг за дефект. Поэтому сначала пришлось научить систему понимать, где находится сама лента. Для этого зафиксировали её ширину — 12 мм. Теперь модель ориентируется на реальные границы ленты, а не на её положение в кадре.
👉 Самыми сложными оказались «плюхи».
У них нет чётких границ. Одна плюха может частично накладываться на другую, а рядом могут лежать отдельные частицы того же материала. Поэтому обычная разметка здесь не работала. Сначала нейросеть искала все точки скопления частиц, а затем объединяла их в группы по заданным правилам. Например, если точки находятся ближе 2 мм друг к другу, система считает их одним дефектом.
👉 Отдельная история — стыки ленты.
Это не брак, а место соединения двух участков ленты. Но на изображениях некоторые стыки выглядели почти так же, как горизонтальные «плюхи». Из-за этого система иногда пропускала реальные дефекты. Пришлось добавить в датасет ещё полсотни примеров таких «плюх» и заново дообучить модель. После этого она перестала путать их со стыками.
Обещали рассказать, как научили нейросеть находить дефекты ленты сверхпроводника и не путать их со сдвигами, стыками и другими допустимыми артефактами производства.
👉 Начнем со сдвигов.
Во время перемотки лента может смещаться в кадре. Для оператора это не проблема, а вот алгоритм легко может принять такой сдвиг за дефект. Поэтому сначала пришлось научить систему понимать, где находится сама лента. Для этого зафиксировали её ширину — 12 мм. Теперь модель ориентируется на реальные границы ленты, а не на её положение в кадре.
👉 Самыми сложными оказались «плюхи».
У них нет чётких границ. Одна плюха может частично накладываться на другую, а рядом могут лежать отдельные частицы того же материала. Поэтому обычная разметка здесь не работала. Сначала нейросеть искала все точки скопления частиц, а затем объединяла их в группы по заданным правилам. Например, если точки находятся ближе 2 мм друг к другу, система считает их одним дефектом.
👉 Отдельная история — стыки ленты.
Это не брак, а место соединения двух участков ленты. Но на изображениях некоторые стыки выглядели почти так же, как горизонтальные «плюхи». Из-за этого система иногда пропускала реальные дефекты. Пришлось добавить в датасет ещё полсотни примеров таких «плюх» и заново дообучить модель. После этого она перестала путать их со стыками.
Итог: датасет на 26 тысяч изображений, точность детекции 95% при минимуме ложных срабатываний.
- 🔥 5
- 👍 1









