CNN - Выберите между двумя классами, показанными на 1 изображении

Question

В настоящее время у меня работает довольно ванильный CNN, но у меня есть несколько изображений, которые содержат «несколько классов», это сортировка изображений цифр по разным классам (не ocr), однако у меня также есть класс под названием «bad_captures», которыйдля изображений, которые не имеют цифр.Однако некоторые изображения имеют некоторые особенности, которые есть у bad_captures, но в то же время они содержат цифры.В настоящее время модель предсказывает эти изображения как плохие, но мне нужно сфокусироваться на цифрах.Любой совет, как это сделать?

Answer 1

Вы не можете явно указать модели, какие функции весить больше.Однако вы можете обучить некоторый простой классификатор, который скажет вам, является ли изображение bad_caputre или нет.Затем вы можете поместить это поверх вашей модели, и если изображение не плохое, то обработайте его.Таким образом, вашей модели CNN не придется беспокоиться о выборе классов, она будет сосредоточена только на получении правильной цифры из изображения.

Другой способ - выяснить, какие из этих функций сбивают модель с толку, и удалитьих явно в предварительной обработке.Кроме того, вы можете попробовать разные вещи в предварительной обработке в целом, чтобы посмотреть, влияет ли это на производительность модели.