Я работаю над проблемой тегирования и аннотации изображений, просто изображение может содержать несколько объектов. Я хочу обучаться inception-v4 для классификации с несколькими метками. Мои данные обучения будут изображением, а вектор длины равен числу классов и имеет по 1 в каждом индексе, если объект существует в изображении. Например, если у меня есть четыре класса (Человек, автомобиль, дерево, здания). Если изображение содержит человека и автомобиль. Тогда мой вектор будет (1, 1, 0, 0).Изменение архитектуры Inception-v4 для выполнения классификации с несколькими метками в Tensorflow
Какие изменения необходимо внести, чтобы обучить inception-v4 для проблемы с меткой и аннотацией?
Нужно ли мне только изменять формат ввода и изменять функцию потерь от softmax до sigmoid_cross_entropy_with_logits в архитектуре начального уровня v4?
https://github.com/tensorflow/models/blob/master/slim/nets/inception_v4.py
Спасибо заранее.
Почему бы вам не попробовать и сообщить нам, как это работает? :) – guinny
как вы строите свой вектор метки? Если вы использовали стандартный по умолчанию стандарт, каждое изображение должно иметь яркую метку стиля в стиле. – neouyghur
это с использованием inception-v3: https://github.com/BartyzalRadek/Multi-label-Inception-net –