解析 tensoflow 预热文件

import tensorflow as tf
from tensorflow_serving.apis import prediction_log_pb2
def par_warmup():
  i = 0
  for serialized_example in tf.compat.v1.io.tf_record_iterator("./assets.extra/tf_serving_warmup_requests"):
    log = prediction_log_pb2.PredictionLog()
    log.ParseFromString(serialized_example)
    print(i, type(serialized_example), type(log), "End")
    i += 1
    print(log)
    break

par_warmup()

predict_log {
  request {
    model_spec {
      name: "model"
      signature_name: "serving_default"
    }
    inputs {
      key: "total_cnt"
      value {
        dtype: DT_INT64
        tensor_shape {
          dim {
            size: 1
          }
          dim {
            size: 1
          }
        }
        int64_val: 1
      }
    }
  }
}