config.pbtxt 988 字节
Newer Older
Y
yunyaoXYY 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57
name: "paddlecls"
platform: "ensemble"
max_batch_size: 16
input [
  {
    name: "INPUT"
    data_type: TYPE_UINT8
    dims: [ -1, -1, 3 ]
  }
]
output [
  {
    name: "CLAS_RESULT"
    data_type: TYPE_STRING
    dims: [ -1 ]
  }
]
ensemble_scheduling {
  step [
    {
      model_name: "preprocess"
      model_version: 1
      input_map {
        key: "preprocess_input"
        value: "INPUT"
      }
      output_map {
        key: "preprocess_output"
        value: "RUNTIME_INPUT"
      }
    },
    {
      model_name: "runtime"
      model_version: 1
      input_map {
        key: "inputs"
        value: "RUNTIME_INPUT"
      }
      output_map {
        key: "save_infer_model/scale_0.tmp_1"
        value: "RUNTIME_OUTPUT"
      }
    },
    {
      model_name: "postprocess"
      model_version: 1
      input_map {
        key: "post_input"
        value: "RUNTIME_OUTPUT"
      }
      output_map {
        key: "post_output"
        value: "CLAS_RESULT"
      }
    }
  ]
}