Несколько меток на одной ограничительной рамке для тензор потока SSD mobilenet

Question

Я настроил SSD mobilenet v1 и ранее также обучал эту модель.Однако в моем наборе данных для каждого ограничивающего прямоугольника есть несколько меток классов.Мой набор данных состоит из лиц, каждое лицо имеет 2 метки: возраст и пол.Обе эти метки имеют одинаковые координаты ограничительной рамки.

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

Можно ли добиться нескольких меток на одном ограничительном боксе с помощью мобильной сети SSD?

Answer 1

Это зависит от реализации, но SSD использует слой softmax для прогнозирования одного класса на ограничивающий прямоугольник, тогда как YOLO прогнозирует индивидуальные оценки достоверности сигмовидной кишки для каждого класса. Таким образом, в SSD выбирается один класс с argmax, но в YOLO вы можете принять несколько классов выше порогового значения.

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