CNN для обнаружения объектов в пределах произвольной границы

Я хочу определить, есть ли объекты в пределах границы.

Граница представляет собой белый прямоугольник, легко идентифицируемый (для человека).

Однако положение границы не является фиксированным.

Объекты маленькие, и обычно только 1 или 2 присутствуют на границе, но видны.

Образцы изображений помечаются только 1, если какой-либо объект находится на границе, и 0, если нет. В частности, у меня нет границы в качестве метки.

Какая хорошая архитектура для классификатора таких изображений? Являются ли слои CNN + MaxPooling моим лучшим выбором?


machine-learning computer-vision conv-neural-network image-recognition
person KalEl    schedule 19.06.2019    source источник
comment
Было бы здорово, если бы вы разместили несколько образцов изображений, чтобы мы могли понять задачу.   -  person GaneshTata    schedule 19.06.2019
comment
К сожалению, я не мог поделиться изображением, потому что это для защищенного приложения. Тем не менее, я могу поделиться тем, что сработало. Resnet50 требовалось слишком много времени для сходимости, к тому же ему требовались хаки, чтобы использовать только большие и глубокие каналы. Архитектура, к которой я прибегла вместо этого, была более мелкой, с тремя башнями, Широкие возможности + Интенсивность + Глубина. Интенсивность и глубина имели идентификатор и сверточные остаточные блоки. Также промежуточно использовал вывод, чтобы сделать обучение еще быстрее. Потребовалось 400 эпох для обучения с нуля, чтобы получить точность 98+% на (сбалансированном) проверочном образце, что довольно хорошо.   -  person KalEl    schedule 13.07.2019
comment
Это потрясающе! Поскольку вы также можете ответить на свой вопрос, было бы здорово, если бы вы могли написать свой комментарий в качестве ответа, чтобы другие знали, что действительно сработало для данной постановки задачи. Благодарю вас!   -  person GaneshTata    schedule 14.07.2019