Обнаружение объектов и подсчет объектов

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

Я хочу использовать api обнаружения объекта в tenorflow, а затем просто переобучить последний слой, чтобы распознать яйца. Дело в том, что у меня уже есть много помеченных изображений, но только с информацией об общем количестве яиц, а не о том, где находятся яйца.

Можно ли как-то переобучить сеть, просто предоставив изображения с соответствующим количеством яиц? Или мне нужно вручную пометить расположение отдельных яиц?

Я думал о том, чтобы решить ее как проблему классификации, в которой разные количества яиц будут разными категориями (0, 1, 2, ... - в общем, никогда не бывает больше 8 яиц). Как вы думаете?

1 ответ

Решение

В итоге я решил использовать API обнаружения объектов в Tensorflow, поэтому мне пришлось вручную отмечать местоположение отдельных яиц на многих изображениях, чтобы переобучить нейронную сеть. Результаты на удивление хорошие.

Вот мое решение, если кто-то столкнется с той же проблемой: https://github.com/havluj/EggDetector.

Другие вопросы по тегам