name: "face_recognition"
platform: "ensemble"
max_batch_size: 16

input [
  {
    name: "input"
    data_type: TYPE_FP32
    dims: [3, 160, 160]
  }
]

output [
  {
    name: "embedding"
    data_type: TYPE_FP32
    dims: [512]
  },
  {
    name: "bbox"
    data_type: TYPE_FP32
    dims: [4]
  },
  {
    name: "score"
    data_type: TYPE_FP32
    dims: [1]
  },
  {
    name: "landmarks"
    data_type: TYPE_FP32
    dims: [5, 2]
  }
]

ensemble_scheduling {
  step {
    model_name: "face_allignment"
    model_version: -1
    input_map { key: "input" value: "input" }
    output_map { key: "bbox" value: "bbox" }
    output_map { key: "score" value: "score" }
    output_map { key: "landmarks" value: "landmarks" }
  }
  step {
    model_name: "face_warp"
    model_version: -1
    input_map { key: "input" value: "input" }
    input_map { key: "landmarks" value: "landmarks" }
    input_map { key: "score" value: "score" }
    output_map { key: "output" value: "embedding" }
  }
}