Fix for Layout and image_info related issues (#10258)
* bugfix78627 * stylefix * fix
This commit is contained in:
@@ -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[] =
|
||||
|
||||
@@ -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);
|
||||
}
|
||||
}
|
||||
|
||||
@@ -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;
|
||||
|
||||
Reference in New Issue
Block a user