IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[OV2.0] Preprocessing api cleanup (#8898)

Browse Source 
* Removed 'inline' Preprocessing API

Even though this API provided a way to specify all pre/post-processing in one line - it was considered inconvinient
With 'getters' API preprocessing code looks more clear for user, so old' inline' API is removed

* Fix pyopenvino build issues

* Update after merged PR#8717
This commit is contained in:
Mikhail Nosov
2021-11-30 12:30:13 +03:00
committed by GitHub
parent f6df0a9c13
commit e2172cd38a
19 changed files with 970 additions and 2018 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
samples/cpp/ngraph_function_creation_sample/main.cpp
View File

@@ -276,23 +276,18 @@ int main(int argc, char* argv[]) {
const Layout tensor_layout{"NHWC"};
// apply preprocessing
// clang-format off
model = ov::preprocess::PrePostProcessor(model)
// 1) InputInfo() with no args assumes a model has a single input
.input(ov::preprocess::InputInfo()
// 2) Set input tensor information:
// - precision of tensor is supposed to be 'u8'
// - layout of data is 'NHWC'
.tensor(ov::preprocess::InputTensorInfo()
.set_layout(tensor_layout)
.set_element_type(element::u8))
// 3) Here we suppose model has 'NCHW' layout for input
.network(ov::preprocess::InputNetworkInfo()
.set_layout("NCHW")))
auto proc = ov::preprocess::PrePostProcessor(model);
// 1) InputInfo() with no args assumes a model has a single input
auto& input_info = proc.input();
// 2) Set input tensor information:
// - layout of data is 'NHWC'
// - precision of tensor is supposed to be 'u8'
input_info.tensor().set_layout(tensor_layout).set_element_type(element::u8);
// 3) Here we suppose model has 'NCHW' layout for input
input_info.network().set_layout("NCHW");
// 4) Once the build() method is called, the preprocessing steps
// for layout and precision conversions are inserted automatically
.build();
// clang-format on
model = proc.build();
// -------- Step 4. Read input images --------