potok/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2026-07-31 08:13:04 +04:00
Code Packages Releases Wiki Activity

Add cfgOutputPrecision

Browse Source
This commit is contained in:
TolyaTalamanov
2022-10-02 20:54:08 +00:00
parent 96844b0ca5
commit b0b77b3047
5 changed files with 212 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/gapi/samples/pipeline_modeling_tool.cpp
+6
View File
@@ -210,6 +210,12 @@ InferParams read<InferParams>(const cv::FileNode& fn) {
params.input_layers = readList<std::string>(fn, "input_layers", name);
params.output_layers = readList<std::string>(fn, "output_layers", name);
params.config = readMap<std::string>(fn["config"]);
auto out_prec_str = readOpt<std::string>(fn["output_precision"]);
if (out_prec_str.has_value()) {
params.out_precision =
cv::optional<int>(strToPrecision(out_prec_str.value()));
}
return params;
}
modules/gapi/samples/pipeline_modeling_tool/pipeline_builder.hpp
+4
View File
@@ -258,6 +258,7 @@ struct InferParams {
std::vector<std::string> input_layers;
std::vector<std::string> output_layers;
std::map<std::string, std::string> config;
cv::util::optional<int> out_precision;
};
class PipelineBuilder {
@@ -362,6 +363,9 @@ void PipelineBuilder::addInfer(const CallParams& call_params,
}
pp->pluginConfig(infer_params.config);
if (infer_params.out_precision) {
pp->cfgOutputPrecision(infer_params.out_precision.value());
}
m_state->networks += cv::gapi::networks(*pp);
addCall(call_params,