конвертировать 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?