diff options
Diffstat (limited to 'source/use_case/img_class/src/UseCaseHandler.cc')
-rw-r--r-- | source/use_case/img_class/src/UseCaseHandler.cc | 4 |
1 files changed, 3 insertions, 1 deletions
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 @@ namespace app { 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(), ' '); |