DeepSpeech/bin/graphdef_binary_to_text.py
2019-08-28 17:53:24 +02:00

21 lines
492 B
Python
Executable File

#!/usr/bin/env python
# -*- coding: utf-8 -*-
import tensorflow.compat.v1 as tfv1
import sys
from google.protobuf import text_format
def main():
# Load and export as string
with tfv1.gfile.FastGFile(sys.argv[1], 'rb') as fin:
graph_def = tfv1.GraphDef()
graph_def.ParseFromString(fin.read())
with tfv1.gfile.FastGFile(sys.argv[1] + 'txt', 'w') as fout:
fout.write(text_format.MessageToString(graph_def))
if __name__ == '__main__':
main()