Двухуровневая классификация с использованием единой модели

Question

Сначала я хочу двоично классифицировать событие «Огонь» в 5000 изображений. Во-вторых, если на изображениях имеется огонь, то классифицируйте далее его городской район (здания) или сельский (как лес). Я использую трансферное обучение с различными моделями, включая VGG16, и настраиваю несколько его последних слоев.

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

Я хочу перенести модель обучения в двоичную классификацию и получить результаты обоих шагов в виде:

img1 пожар сельский / городской 1 Нет огня Нет сельских / нет городских 2 Огонь городской 3 Пожар в деревне

так что я могу переобучить vgg16 таким образом, чтобы обеспечить как уровень классификации i-е, шаг первый (пожар / нет), так и шаг второй (сельский / городской)

Answer 1

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