конвертировать save_model.pbtxt в save_model.pb в Tensorflow 2

Я использовал этот скрипт для преобразования файла save_model.pb в файл save_model.pbtxt. Затем я изменил некоторые параметры (is_training: b: True). Теперь я хотел бы преобразовать обратно в save_model.pb, чтобы его можно было прочитать в tf2onnx.

      def convert_saved_model_to_pbtxt(path):
    import os, sys
    import google.protobuf
    from tensorflow.core.protobuf import saved_model_pb2
    saved_model = saved_model_pb2.SavedModel()
    with open(os.path.join(path, 'saved_model.pb'), 'rb') as f:
        saved_model.ParseFromString(f.read())
    with open(os.path.join(path, 'saved_model.pbtxt'), 'w') as f:
        f.write(google.protobuf.text_format.MessageToString(saved_model))

Каков рекомендуемый способ преобразования этого .pbtxt в ONNX?

0 ответов

Другие вопросы по тегам