1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
|
/*
* Copyright (c) 2021-2022 Arm Limited. All rights reserved.
* SPDX-License-Identifier: Apache-2.0
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#ifndef USECASE_COMMON_UTILS_HPP
#define USECASE_COMMON_UTILS_HPP
#include "hal.h"
#include "Model.hpp"
#include "AppContext.hpp"
#include "Profiler.hpp"
#include "UseCaseHandler.hpp" /* Handlers for different user options. */
#include "Classifier.hpp" /* Classifier. */
#include "InputFiles.hpp"
#include <cinttypes>
/* Helper macro to convert RGB888 to RGB565 format. */
#define RGB888_TO_RGB565(R8,G8,B8) ((((R8>>3) & 0x1F) << 11) | \
(((G8>>2) & 0x3F) << 5) | \
((B8>>3) & 0x1F))
constexpr uint16_t COLOR_BLACK = 0;
constexpr uint16_t COLOR_GREEN = RGB888_TO_RGB565( 0u, 255u, 0u); // 2016;
constexpr uint16_t COLOR_YELLOW = RGB888_TO_RGB565(255u, 255u, 0u); // 65504;
void DisplayCommonMenu();
namespace image{
/**
* @brief Helper function to convert a UINT8 image to INT8 format.
* @param[in,out] data Pointer to the data start.
* @param[in] kMaxImageSize Total number of pixels in the image.
**/
void ConvertImgToInt8(void * data, size_t kMaxImageSize);
/**
* @brief Presents inference results using the data presentation
* object.
* @param[in] platform Reference to the hal platform object.
* @param[in] results Vector of classification results to be displayed.
* @return true if successful, false otherwise.
**/
bool PresentInferenceResult(hal_platform & platform,
const std::vector < arm::app::ClassificationResult > & results);
/**
* @brief Converts RGB image to grayscale.
* @param[in] srcPtr Pointer to RGB source image.
* @param[out] dstPtr Pointer to grayscale destination image.
* @param[in] imgSz Destination image size.
**/
void RgbToGrayscale(const uint8_t *srcPtr, uint8_t *dstPtr, const size_t dstImgSz);
}
/**
* @brief Helper function to increment current input feature vector index.
* @param[in,out] ctx Pointer to the application context object.
* @param[in] useCase Use case name
**/
void IncrementAppCtxIfmIdx(arm::app::ApplicationContext& ctx, const std::string& useCase);
/**
* @brief Helper function to set the input feature map index.
* @param[in,out] ctx Pointer to the application context object.
* @param[in] idx Value to be set.
* @param[in] ctxIfmName Input Feature Map name
* @return true if index is set, false otherwise.
**/
bool SetAppCtxIfmIdx(arm::app::ApplicationContext& ctx, uint32_t idx, const std::string& ctxIfmName);
namespace common {
enum OPCODES {
MENU_OPT_RUN_INF_NEXT = 1, /* Run on next vector. */
MENU_OPT_RUN_INF_CHOSEN, /* Run on a user provided vector index. */
MENU_OPT_RUN_INF_ALL, /* Run inference on all. */
MENU_OPT_SHOW_MODEL_INFO, /* Show model info. */
MENU_OPT_LIST_IFM /* List the current IFM. */
};
}
namespace arm {
namespace app {
/**
* @brief Run inference using given model
* object. If profiling is enabled, it will log the
* statistics too.
* @param[in] model Reference to the initialised model.
* @param[in] profiler Reference to the initialised profiler.
* @return true if inference succeeds, false otherwise.
**/
bool RunInference(arm::app::Model& model, Profiler& profiler);
/**
* @brief Read input and return as an integer.
* @param[in] platform Reference to the hal platform object.
* @return Integer value corresponding to the user input.
**/
int ReadUserInputAsInt(hal_platform& platform);
#if VERIFY_TEST_OUTPUT
/**
* @brief Helper function to dump a tensor to stdout
* @param[in] tensor tensor to be dumped
* @param[in] lineBreakForNumElements number of elements
* after which line break will be added.
**/
void DumpTensor(const TfLiteTensor* tensor,
size_t lineBreakForNumElements = 16);
void DumpTensorData(const uint8_t* tensorData,
size_t size,
size_t lineBreakForNumElements = 16);
#endif /* VERIFY_TEST_OUTPUT */
/**
* @brief List the files baked in the application.
* @param[in] ctx Reference to the application context.
* @return true or false based on event being handled.
**/
bool ListFilesHandler(ApplicationContext& ctx);
} /* namespace app */
} /* namespace arm */
#endif /* USECASE_COMMON_UTILS_HPP */
|