diff options
Diffstat (limited to 'source/use_case/img_class/src')
-rw-r--r-- | source/use_case/img_class/src/MobileNetModel.cc | 16 | ||||
-rw-r--r-- | source/use_case/img_class/src/UseCaseHandler.cc | 2 |
2 files changed, 8 insertions, 10 deletions
diff --git a/source/use_case/img_class/src/MobileNetModel.cc b/source/use_case/img_class/src/MobileNetModel.cc index eeaa109..b937382 100644 --- a/source/use_case/img_class/src/MobileNetModel.cc +++ b/source/use_case/img_class/src/MobileNetModel.cc @@ -20,20 +20,20 @@ const tflite::MicroOpResolver& arm::app::MobileNetModel::GetOpResolver() { - return this->_m_opResolver; + return this->m_opResolver; } bool arm::app::MobileNetModel::EnlistOperations() { - this->_m_opResolver.AddDepthwiseConv2D(); - this->_m_opResolver.AddConv2D(); - this->_m_opResolver.AddAveragePool2D(); - this->_m_opResolver.AddAdd(); - this->_m_opResolver.AddReshape(); - this->_m_opResolver.AddSoftmax(); + this->m_opResolver.AddDepthwiseConv2D(); + this->m_opResolver.AddConv2D(); + this->m_opResolver.AddAveragePool2D(); + this->m_opResolver.AddAdd(); + this->m_opResolver.AddReshape(); + this->m_opResolver.AddSoftmax(); #if defined(ARM_NPU) - if (kTfLiteOk == this->_m_opResolver.AddEthosU()) { + if (kTfLiteOk == this->m_opResolver.AddEthosU()) { info("Added %s support to op resolver\n", tflite::GetString_ETHOSU()); } else { diff --git a/source/use_case/img_class/src/UseCaseHandler.cc b/source/use_case/img_class/src/UseCaseHandler.cc index fa77512..337cb29 100644 --- a/source/use_case/img_class/src/UseCaseHandler.cc +++ b/source/use_case/img_class/src/UseCaseHandler.cc @@ -58,8 +58,6 @@ namespace app { * object. * @param[in] platform Reference to the hal platform object. * @param[in] results Vector of classification results to be displayed. - * @param[in] infTimeMs Inference time in milliseconds, if available - * otherwise, this can be passed in as 0. * @return true if successful, false otherwise. **/ static bool PresentInferenceResult(hal_platform& platform, |