diff options
author | Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> | 2019-01-24 17:05:36 +0000 |
---|---|---|
committer | Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> | 2019-01-30 11:25:56 +0000 |
commit | 7cf0eaa26c1fb29ca9df97e4734ec7c1e10f81c4 (patch) | |
tree | 407b519ede76b235c54907fe80411970741e8a00 /tests/CaffeYolo-Armnn | |
parent | 28d3d63cc0a33f8396b32fa8347c03912c065911 (diff) | |
download | armnn-7cf0eaa26c1fb29ca9df97e4734ec7c1e10f81c4.tar.gz |
IVGCVSW-2564 Add support for multiple input and output bindings in InferenceModel
Change-Id: I64d724367d42dca4b768b6c6e42acda714985950
Diffstat (limited to 'tests/CaffeYolo-Armnn')
-rw-r--r-- | tests/CaffeYolo-Armnn/CaffeYolo-Armnn.cpp | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/tests/CaffeYolo-Armnn/CaffeYolo-Armnn.cpp b/tests/CaffeYolo-Armnn/CaffeYolo-Armnn.cpp index 98db023e97..b752c7c98e 100644 --- a/tests/CaffeYolo-Armnn/CaffeYolo-Armnn.cpp +++ b/tests/CaffeYolo-Armnn/CaffeYolo-Armnn.cpp @@ -31,9 +31,9 @@ int main(int argc, char* argv[]) typename YoloInferenceModel::Params modelParams; modelParams.m_ModelPath = modelOptions.m_ModelDir + "yolov1_tiny_voc2007_model.caffemodel"; - modelParams.m_InputBinding = "data"; - modelParams.m_OutputBinding = "fc12"; - modelParams.m_InputTensorShape = &inputTensorShape; + modelParams.m_InputBindings = { "data" }; + modelParams.m_OutputBindings = { "fc12" }; + modelParams.m_InputShapes = { inputTensorShape }; modelParams.m_IsModelBinary = true; modelParams.m_ComputeDevice = modelOptions.m_ComputeDevice; modelParams.m_VisualizePostOptimizationModel = modelOptions.m_VisualizePostOptimizationModel; |