Узлы модели классификации изображений TF-SLIM отличаются от заданных предварительно обученных сетевых моделей - PullRequest
0 голосов
/ 24 февраля 2020

Я обучил классификации изображений модель MobileNetV2, используя модель в https://github.com/tensorflow/models/tree/master/research/slim/nets/mobilenet

Однако, когда я конвертирую свои контрольные точки в замороженный график (.pb) и проверяю узлы в тензорной доске узлы отличаются от заданной модели MobileNetV2 с предварительной подготовкой Imag eNet по той же ссылке выше (см. файл README).

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

Pretrained MobileNetV2

My Trained MobileNetV2

Как видно выше, моя обученная модель имеет дополнительный узел fifo_queue. У него есть и другие дополнительные узлы. Я хочу, чтобы у моей обученной модели были узлы, точно такие же, как у предварительно обученного, потому что дополнительные узлы не могут быть преобразованы в .uff, а TensorRT не может проанализировать эти узлы.

Предупреждения, которые я получаю, когда преобразовываю граф .pb в .uff : pb to uff

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

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

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