Fix for Layout and image_info related issues (#10258)

* bugfix78627

* stylefix

* fix
This commit is contained in:
Fedor Zharinov
2022-02-17 00:42:51 +03:00
committed by GitHub
parent 5cadee20eb
commit 00abcbacc4
3 changed files with 12 additions and 3 deletions

View File

@@ -24,7 +24,10 @@ static const char input_message[] =
" of files for each input (except cases with single file for any input):"
"\"input1:1.jpg input2:1.bin\", \"input1:1.bin,2.bin input2:3.bin input3:4.bin,5.bin \"."
" Also you can pass specific keys for inputs: \"random\" - for fillling input with random data,"
" \"image_info\" - for filling input with image size.";
" \"image_info\" - for filling input with image size.\n"
" You should specify either one files set to be used for all inputs (without "
"providing "
"input names) or separate files sets for every input of model (providing inputs names).";
/// @brief message for model argument
static const char model_message[] =

View File

@@ -550,7 +550,9 @@ int main(int argc, char* argv[]) {
info.at(name).type = type_to_set;
}
}
// Explicitly set inputs layout.
}
// Explicitly set inputs layout.
if (!name.empty() && !app_inputs_info[0].at(name).layout.empty()) {
in.model().set_layout(app_inputs_info[0].at(name).layout);
}
}

View File

@@ -317,7 +317,11 @@ std::map<std::string, std::vector<std::string>> parse_input_arguments(const std:
}
for (auto& file : files.second) {
readInputFilesArguments(mapped_files[files.first], file);
if (file == "image_info" || file == "random") {
mapped_files[files.first].push_back(file);
} else {
readInputFilesArguments(mapped_files[files.first], file);
}
}
}
args_it = files_end;