Add clamp functionality to PostProcessSteps for value range control

Author: barnasm1

src/core/include/openvino/core/preprocess/postprocess_steps.hpp

@@ -33,6 +33,8 @@ class OPENVINO_API PostProcessSteps final {
     /// \brief Default destructor
     ~PostProcessSteps();
 
+    PostProcessSteps& clamp(double min_value, double max_value);
+
     /// \brief Add convert element type post-process operation
     ///
     /// \param type Desired type of output. If not specified, type will be obtained from 'tensor' output information

src/core/src/preprocess/pre_post_process.cpp

@@ -402,6 +402,11 @@ OutputModelInfo& OutputModelInfo::set_color_format(const ov::preprocess::ColorFo
 PostProcessSteps::PostProcessSteps() : m_impl(std::unique_ptr<PostProcessStepsImpl>(new PostProcessStepsImpl())) {}
 PostProcessSteps::~PostProcessSteps() = default;
 
+PostProcessSteps& PostProcessSteps::clamp(double min_value, double max_value) {
+    m_impl->add_clamp(min_value, max_value);
+    return *this;
+}
+
 PostProcessSteps& PostProcessSteps::convert_element_type(const element::Type& type) {
     m_impl->add_convert_impl(type);
     return *this;

src/core/src/preprocess/preprocess_steps_impl.cpp

@@ -712,6 +712,23 @@ std::tuple<std::vector<Output<Node>>, bool> PreStepsList::cut_last_channel(const
 }
 
 //------------- Post processing ------
+void PostStepsList::add_clamp(double min_value, double max_value) {
+    std::string name = "clamp(min " + std::to_string(min_value) + ", max " + std::to_string(max_value) + ")";
+
+    m_actions.emplace_back(
+        [min_value, max_value](const Output<Node>& node, PostprocessingContext& ctxt) {
+            auto element_type = node.get_element_type();
+            OPENVINO_ASSERT(element_type.is_real(),
+                            "Clamp postprocessing can be applied to 'double' inputs. Consider using "
+                            "'convert_element_type' before clamping. Current type is: ",
+                            element_type);
+
+            auto clamp_op = std::make_shared<ov::op::v0::Clamp>(node, min_value, max_value);
+            return std::make_tuple(Output<Node>{clamp_op}, true);
+        },
+        name);
+}
+
 void PostStepsList::add_convert_impl(const element::Type& type) {
     m_actions.emplace_back(
         [type](const Output<Node>& node, PostprocessingContext& ctxt) {

src/core/src/preprocess/preprocess_steps_impl.hpp

@@ -221,6 +221,7 @@ struct InternalPostprocessAction {
 /// \brief PostProcessStepsImpl - internal data structure
 class PostStepsList {
 public:
+    void add_clamp(double min_value, double max_value);
     void add_convert_impl(const element::Type& type);
     void add_convert_layout_impl(const Layout& layout);
     void add_convert_layout_impl(const std::vector<uint64_t>& dims);