Я обучил классификации изображений модель MobileNetV2, используя модель в https://github.com/tensorflow/models/tree/master/research/slim/nets/mobilenet
Однако, когда я конвертирую свои контрольные точки в замороженный график (.pb) и проверяю узлы в тензорной доске узлы отличаются от заданной модели MobileNetV2 с предварительной подготовкой Imag eNet по той же ссылке выше (см. файл README).
Я загружаю два примера рисунков, показывающих входные узлы обученной и предварительно обученной моделей.
Как видно выше, моя обученная модель имеет дополнительный узел fifo_queue
. У него есть и другие дополнительные узлы. Я хочу, чтобы у моей обученной модели были узлы, точно такие же, как у предварительно обученного, потому что дополнительные узлы не могут быть преобразованы в .uff, а TensorRT не может проанализировать эти узлы.
Предупреждения, которые я получаю, когда преобразовываю граф .pb в .uff :
Как я могу убедиться, что моя обученная модель имеет те же узлы, что и предоставленная предварительно обученная модель MobileNetV2?
Я пытался обучить свою сеть с помощью клонирования репозиторий тензорного потока с того момента, когда MobileNetV2 был зафиксирован впервые, но эта модель также имела такие же дополнительные узлы.