Edit this page

class ov::preprocess::InputModelInfo

Overview

Information about model’s input tensor. If all information is already included to loaded model, this info may not be needed. However it can be set to specify additional information about model, like ‘layout’. More…

#include <input_model_info.hpp>

class InputModelInfo
{
public:
    // methods

    InputModelInfo& set_layout(const ov::Layout& layout);
};

Detailed Documentation

Information about model’s input tensor. If all information is already included to loaded model, this info may not be needed. However it can be set to specify additional information about model, like ‘layout’.

Example of usage of model ‘layout’: Support model has input parameter with shape {1, 3, 224, 224} and user needs to resize input image to model’s dimensions. It can be done like this

<model has input parameter with shape {1, 3, 224, 224}>
auto proc = PrePostProcessor(function);
proc.input().preprocess().resize(ResizeAlgorithm::RESIZE_LINEAR);
proc.input().model().set_layout("NCHW");

Methods

InputModelInfo& set_layout(const ov::Layout& layout)

Set layout for model’s input tensor This version allows chaining for Lvalue objects.

Parameters:

layout

Layout for model’s input tensor.

Returns:

Reference to ‘this’ to allow chaining with other calls in a builder-like manner

Prev Next