Update dnn_benchmark for to get size

JDAI-CV · May 13, 2019 · cced7ed · cced7ed
1 parent 3d21350
commit cced7ed
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/binaries/dnn_benchmark.cpp b/binaries/dnn_benchmark.cpp
@@ -70,8 +70,8 @@ int main(int argc, char **argv) {
     {
         auto model = get_model(daqName, outputBlob, false,
                                ANEURALNETWORKS_PREFER_FAST_SINGLE_ANSWER);
-        inputLen = model->GetInputSize(0);
-        outputLen = model->GetOutputSize(0);
+        inputLen = model->GetSize(model->GetInputs()[0]);
+        outputLen = model->GetSize(model->GetOutputs()[0]);
     }
 #define WARM_UP                                                            \
     {                                                                      \