Как дать несколько ярлыков с изображением в тензорном потоке?

Question

Я хочу implememt многозадачной нейронной сети в tensorflow, для которого мне нужен мой вход как:

[Изображение label1 Метка2]

, который я могу дать в нейронную сеть для обучения.

Мой вопрос в том, как я могу связать несколько ярлыков с изображением в файле TFRecord?

Я в настоящее время использовал build_image_data.py файл исходной модели для генерации файла TFRecord, но в этом случае есть только одна метка на изображение.

Answer 1

Есть ли у них всегда две этикетки? Если да, попробуйте «label1-label2» в качестве одной метки. Или просто постройте две сети, одну для метки 1, а другую для этикетки 2.

Являются ли они иерархическими ярлыками? Затем проверьте иерархические классификаторы.

Answer 2

1. Я получил эту работу. Для любого, кто ищет ссылку, вы можете изменить пример proto файла build_image_data.py и связать его двумя ярлыками.)