commit 27b62d9dec86b76422c6c8be7615c1f712518570
author alexander <alexander.efremov@arm.com> Tue May 04 20:46:08 2021 +0100
committer Isabella Gottardi <isabella.gottardi@arm.com> Wed May 05 10:58:09 2021 +0000
tree c0bc0d805f1a88e1d92619b2bb50572b688ec126
parent c350cdced0a8a2ca17376f58813e6d48d796ac7c

If inference fails - exit before printing results.
It was easy to miss error message and mistakenly assume that execution succeeded.

Signed-off-by: alexander <alexander.efremov@arm.com>
Change-Id: I3c7beb5afe97db5df486e12e3bfe0e8e27e382f8

source/use_case/img_class/src/UseCaseHandler.cc

diff --git a/source/use_case/img_class/src/UseCaseHandler.cc b/source/use_case/img_class/src/UseCaseHandler.cc
index ffeb860..22e6ba0 100644
--- a/source/use_case/img_class/src/UseCaseHandler.cc
+++ b/source/use_case/img_class/src/UseCaseHandler.cc
@@ -145,7 +145,9 @@
             info("Running inference on image %u => %s\n", ctx.Get<uint32_t>("imgIndex"),
                 get_filename(ctx.Get<uint32_t>("imgIndex")));
 
-            RunInference(model, profiler);
+            if (!RunInference(model, profiler)) {
+                return false;
+            }
 
             /* Erase. */
             str_inf = std::string(str_inf.size(), ' ');