Tensorflow: загрузка предварительно обученных весов модели в другую архитектуру - PullRequest
0 голосов
/ 09 июля 2019

У меня есть модель Tensorflow, которая достаточно хорошо работает для обнаружения объекта на изображении и создания ограничивающего прямоугольника.Выход включает один Softmax и 4 аналоговых значения для местоположения.Мне нужно добавить еще один аналоговый выход для прогнозирования ориентации объекта.Как я могу импортировать предварительно обученные веса моделей и заморозить их так, чтобы обучалась только часть, имеющая отношение к ориентации в последнем слое.

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