name: "od"
platform: "ensemble"

input [
    { 
        name: "IMAGE_BYTES"
        data_type: TYPE_STRING
        dims: [1]
    }
]

output [
    { 
        name: "DETECTIONS" 
        data_type: TYPE_FP32 
        dims: [1, -1, 4] 
    },
    { 
        name: "ORIGINAL_SHAPE" 
        data_type: TYPE_UINT16
        dims: [2] 
    },
    { 
        name: "RESIZED_SHAPE" 
        data_type: TYPE_UINT16
        dims: [2] 
    }
]

ensemble_scheduling {
    step {
        model_name: "od_preprocess"
        model_version: -1

        input_map { key: "image_bytes" value: "IMAGE_BYTES" }

        output_map { key: "resized_image" value: "resized_image" },
        output_map { key: "original_shape" value: "ORIGINAL_SHAPE" },
        output_map { key: "resized_shape" value: "RESIZED_SHAPE" }
    }
    step {
        model_name: "od_engine"
        model_version: -1

        input_map { key: "images" value: "resized_image" }
        
        output_map { key: "output0" value: "model_output" }
    }
    step {
        model_name: "od_postprocess"
        model_version: -1

        input_map { key: "model_output" value: "model_output" }

        output_map { key: "detections" value: "DETECTIONS" }
    }
}