From 0267cbd286fd945799ece06581d61f29b64296c4 Mon Sep 17 00:00:00 2001 From: Mikhail Ryzhov Date: Mon, 26 Oct 2020 15:36:49 +0300 Subject: [PATCH] Fixed ngraph sample logical errors (#2600) * Fixed ngrapth sample logical errors - added batch size, because sample accepts several images - set label file as optional --- inference-engine/samples/ngraph_function_creation_sample/main.cpp | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/inference-engine/samples/ngraph_function_creation_sample/main.cpp b/inference-engine/samples/ngraph_function_creation_sample/main.cpp index b360c08..10757bb 100644 --- a/inference-engine/samples/ngraph_function_creation_sample/main.cpp +++ b/inference-engine/samples/ngraph_function_creation_sample/main.cpp @@ -272,7 +272,8 @@ int main(int argc, char* argv[]) { } /** Setting batch size using image count **/ - size_t batchSize = 1; + network.setBatchSize(imagesData.size()); + size_t batchSize = network.getBatchSize(); slog::info << "Batch size is " << std::to_string(batchSize) << slog::endl; // --------------------------- Prepare output blobs ----------------------------------------------------- @@ -383,8 +384,6 @@ int main(int argc, char* argv[]) { labels.push_back(strLine); } inputFile.close(); - } else { - throw std::logic_error("Cannot read label file"); } ClassificationResult classificationResult(outputBlob, images, batchSize, FLAGS_nt, labels); -- 2.7.4