ML модель для распознавания брака на радиоэлектронном производстве (для всех)

На входе есть снимки с электронного микроскопа, есть ряд категорий брака (плохая пропайка и пр) которые можно визуально идентифицировать. Нужна модель на Keras