Como usar a ferramenta freeze_graph.py no TensorFlow v1
É possível usar ofreeze_graph.py
ferramenta com modelos salvos viasaver.save
no TensorFlow v1? Se sim, como?
Eu tenho um código que se parece com isso:
supervisor = tf.train.Supervisor(logdir=output_directory_path)
with supervisor.managed_session() as session:
# train the model here
supervisor.saver.save(session, output_directory_path)
Isso produz um diretório que contém:
checkpoint
output
output-16640.data-00000-of-00001
,output-16640.index
output-16640.meta
Ondeoutput
é um diretório que contém os arquivos para as etapas intermediárias de treinamento. O resto são arquivos.
Meu entendimento é que este é um meta-gráfico (o.meta
arquivo) e suas variáveis (o.data*
arquivo) no formato saver v2. Esses arquivos contêm os dados necessários para ofreeze_graph.py
ferramenta, mas não está claro como saberfreeze_graph.py
ferramenta para carregar os dados desses arquivos.
Todas essas tentativas produzem a mensagem de erroInput checkpoint '...' doesn't exist!
python freeze_graph.py --input_checkpoint checkpoint --output_graph /tmp/out
python freeze_graph.py --input_checkpoint . --output_graph /tmp/out
python freeze_graph.py --input_checkpoint output-16640 --output_graph /tmp/out
ofreeze_graph.py
código inclui o comentário'input_checkpoint' may be a prefix if we're using Saver V2 format
próximo a onde o--input_checkpoint
argumento é usado, então eu pensei que a terceira das tentativas acima funcionaria, mas, infelizmente, não.