Как преобразовать файл `.pb` в`.dlc` с помощью модели, обученной с `tf.data.Dataset`?
Я тренировал модель с tf.data.Dataset в качестве ввода. И файл контрольной точки был заморожен .pb файл с freeze_graph.py. Первые три узла в .pb показаны ниже:
['OneShotIterator',
'IteratorGetNext',
'holi/conv2d/kernel',
...]
OneShotIterator является входным узлом из-за tf.data.Dataset, который не имеет формы. Но snpe-tensorflow-to-dlc требует формы входного узла.
Я думаю, что tf.data.Dataset хороший способ подачи данных, но кажется, что его трудно использовать OneShotIterator в качестве модели ввода.
Мой код организован с estimator а также datasetи это огромная работа, чтобы изменить метод ввода данных на feed,
Любая идея сделать входной узел с явной формой?