diff options
author | Richard Burton <richard.burton@arm.com> | 2022-04-19 17:01:08 +0100 |
---|---|---|
committer | Richard Burton <richard.burton@arm.com> | 2022-04-19 17:01:08 +0100 |
commit | c20be97af1c3e4d569d37587cd22d343193c7563 (patch) | |
tree | 77c6baa0e2024ab63cc89df3763a2a3dfa25047a /source/use_case/img_class/src/UseCaseHandler.cc | |
parent | ea8ce56630544600b112d24e6bf51307fcbb93ae (diff) | |
download | ml-embedded-evaluation-kit-c20be97af1c3e4d569d37587cd22d343193c7563.tar.gz |
MLECO-3078: Add VWW use case API
We now expect that img_class and vww use cases model inputs should have 4 dims as dictated by
use case logic
Signed-off-by: Richard Burton <richard.burton@arm.com>
Change-Id: I67a57a3a28a7ff2e09c917c40e9fc2c08384a45c
Diffstat (limited to 'source/use_case/img_class/src/UseCaseHandler.cc')
-rw-r--r-- | source/use_case/img_class/src/UseCaseHandler.cc | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/source/use_case/img_class/src/UseCaseHandler.cc b/source/use_case/img_class/src/UseCaseHandler.cc index 11a1aa8..c68d816 100644 --- a/source/use_case/img_class/src/UseCaseHandler.cc +++ b/source/use_case/img_class/src/UseCaseHandler.cc @@ -43,7 +43,7 @@ namespace app { return false; } } - auto initialImIdx = ctx.Get<uint32_t>("imgIndex"); + auto initialImgIdx = ctx.Get<uint32_t>("imgIndex"); constexpr uint32_t dataPsnImgDownscaleFactor = 2; constexpr uint32_t dataPsnImgStartX = 10; @@ -62,8 +62,8 @@ namespace app { if (!inputTensor->dims) { printf_err("Invalid input tensor dims\n"); return false; - } else if (inputTensor->dims->size < 3) { - printf_err("Input tensor dimension should be >= 3\n"); + } else if (inputTensor->dims->size < 4) { + printf_err("Input tensor dimension should be = 4\n"); return false; } @@ -148,7 +148,7 @@ namespace app { IncrementAppCtxIfmIdx(ctx,"imgIndex"); - } while (runAll && ctx.Get<uint32_t>("imgIndex") != initialImIdx); + } while (runAll && ctx.Get<uint32_t>("imgIndex") != initialImgIdx); return true; } |