[GNA][Speech sample] Add option to specify blob names (#1529)

* Added output names

* Add input, output, ref names

* Added zero scale factor

* Adding support for multiple reference files
This commit is contained in:
Andrey Dmitriev
2020-10-16 15:34:22 +03:00
committed by GitHub
parent 95d7c29628
commit aff7a66082
2 changed files with 363 additions and 245 deletions

View File

@@ -81,6 +81,16 @@ static const char context_window_message_r[] = "Optional. Number of frames for r
"Works only with context window networks."
" If you use the cw_r or cw_l flag, then batch size and nthreads arguments are ignored.";
/// @brief message for output layer names
static const char output_layer_names_message[] = "Optional. Layer names for output blobs. " \
"The names are separated with \",\" " \
"Example: input1:port,input2:port ";
/// @brief message for inputs layer names
static const char input_layer_names_message[] = "Optional. Layer names for input blobs. " \
"The names are separated with \",\" " \
"Example: Input1,Input2 ";
/// \brief Define flag for showing help message <br>
DEFINE_bool(h, false, help_message);
@@ -145,6 +155,12 @@ DEFINE_int32(cw_r, 0, context_window_message_r);
/// @brief Left context window size (default 0)
DEFINE_int32(cw_l, 0, context_window_message_l);
/// @brief Output layer name
DEFINE_string(oname, "", output_layer_names_message);
/// @brief Input layer name
DEFINE_string(iname, "", input_layer_names_message);
/**
* \brief This function show a help message
*/
@@ -173,5 +189,7 @@ static void showUsage() {
std::cout << " -nthreads \"<integer>\" " << infer_num_threads_message << std::endl;
std::cout << " -cw_l \"<integer>\" " << context_window_message_l << std::endl;
std::cout << " -cw_r \"<integer>\" " << context_window_message_r << std::endl;
std::cout << " -oname \"<string>\" " << output_layer_names_message << std::endl;
std::cout << " -iname \"<string>\" " << input_layer_names_message << std::endl;
}