diff options
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; |