Как использовать модель CNN для обнаружения объекта, распознанного YOLO - PullRequest
0 голосов
/ 18 июня 2019

Давайте начнем с того, что у меня есть 2 предварительно обученные модели (в файлах hdf5):

Первая модель - это модель на основе YOLO, обученная на наборе данных A, которая используется для определения местоположения человека на любых изображениях (обратите внимание: обученные изображения на этой модели могут содержать много людей внутри)

Вторая модель - это модель CNN, которая используется для определения пола человека (мужчины или женщины) на основе изображения, содержащего только 1 человека.

Предположим, что я хочу использовать только эти 2 модели и не хочу переобучать или изменять что-либо в наборе данных. Как я могу найти женщину на снимке Dataset A?

Возможное решение, которое, я думаю, могло бы работать:

Сначала используйте первую модель для обнаружения, то есть для создания ограничивающих рамок вокруг людей на изображениях.

Обрежьте ограничительные рамки в уникальные изображения. Подайте эти изображения второй модели, чтобы увидеть, является ли этот человек женщиной / мужчиной

Однако это решение работает медленно. Так есть ли в любом случае, которые могут закрепить это решение или выполнить эту задачу различными способами?

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...