54 lines
986 B (Stored with Git LFS)
Plaintext
54 lines
986 B (Stored with Git LFS)
Plaintext
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" }
|
|
}
|
|
}
|