Background subtraction

Author: DariaMityagina

demos/background_subtraction_demo/cpp_gapi/include/custom_kernels.hpp

@@ -49,10 +49,10 @@ class NNBGReplacer {
     }
 
 protected:
-    IE::CNNNetwork m_cnn_network;
+    std::shared_ptr<ov::Model> m_model;
     std::string m_tag;
-    IE::InputsDataMap m_inputs;
-    IE::OutputsDataMap m_outputs;
+    std::vector<ov::Output<ov::Node>> m_inputs;
+    std::vector<ov::Output<ov::Node>> m_outputs;
 };
 
 class MaskRCNNBGReplacer : public NNBGReplacer {

demos/background_subtraction_demo/cpp_gapi/src/custom_kernels.cpp

@@ -22,6 +22,8 @@
 #include <opencv2/gapi/own/assert.hpp>
 #include <opencv2/imgproc.hpp>
 
+#include "openvino/openvino.hpp"
+
 // clang-format off
 GAPI_OCV_KERNEL(OCVTensorToImg, custom::GTensorToImg) {
     static void run(const cv::Mat &in,
@@ -137,26 +139,26 @@ GAPI_OCV_KERNEL(OCVCalculateMaskRCNNBGMask, custom::GCalculateMaskRCNNBGMask) {
 // clang-format on
 
 custom::NNBGReplacer::NNBGReplacer(const std::string& model_path) {
-    IE::Core core;
-    m_cnn_network = core.ReadNetwork(model_path);
-    m_tag = m_cnn_network.getName();
-    m_inputs = m_cnn_network.getInputsInfo();
-    m_outputs = m_cnn_network.getOutputsInfo();
+    ov::Core core;
+    m_model = core.read_model(model_path);
+    m_tag = m_model->get_name();
+    m_inputs = m_model->inputs();
+    m_outputs = m_model->outputs();
 }
 
 custom::MaskRCNNBGReplacer::MaskRCNNBGReplacer(const std::string& model_path) : custom::NNBGReplacer(model_path) {
     for (const auto& p : m_outputs) {
-        const auto& layer_name = p.first;
+        const auto& layer_name = p.get_node()->get_name();
         if (layer_name.rfind("TopK") != std::string::npos) {
             continue;
         }
 
         if (m_inputs.size() != 1) {
             throw std::logic_error("Supported only single input MaskRCNN models!");
         }
-        m_input_name = m_inputs.begin()->first;
+        m_input_name = m_inputs.begin()->get_node()->get_name();
 
-        const auto dims_size = p.second->getTensorDesc().getDims().size();
+        const auto dims_size = p.get_tensor().get_partial_shape().size();
         if (dims_size == 1) {
             m_labels_name = layer_name;
         } else if (dims_size == 2) {
@@ -177,7 +179,14 @@ cv::GMat custom::MaskRCNNBGReplacer::replace(cv::GFrame in, cv::GMat bgr, const
     auto boxes = outputs.at(m_boxes_name);
     auto masks = outputs.at(m_masks_name);
 
-    const auto& dims = m_inputs.at(m_input_name)->getTensorDesc().getDims();
+    std::map<std::string, ov::Output<ov::Node>> inputs_map;
+    for (auto in : m_inputs) {
+        std::cout << "in.get_node()->get_name() = " << in.get_node()->get_name() << "\n";
+        inputs_map[in.get_node()->get_name()] = in;
+    }
+    std::cout << "m_input_name = " << m_input_name << "\n";
+
+    const auto& dims = inputs_map.at(m_input_name).get_tensor().get_partial_shape().get_max_shape();
     GAPI_Assert(dims.size() == 4u);
     auto mask = custom::GCalculateMaskRCNNBGMask::on(in_size, cv::Size(dims[3], dims[2]), labels, boxes, masks);
     auto mask3ch = cv::gapi::medianBlur(cv::gapi::merge3(mask, mask, mask), 11);
@@ -188,12 +197,12 @@ custom::BGMattingReplacer::BGMattingReplacer(const std::string& model_path) : NN
     if (m_inputs.size() != 1) {
         throw std::logic_error("Supported only single input background matting models!");
     }
-    m_input_name = m_inputs.begin()->first;
+    m_input_name = m_inputs.begin()->get_node()->get_name();
 
     if (m_outputs.size() != 1) {
         throw std::logic_error("Supported only single output background matting models!");
     }
-    m_output_name = m_outputs.begin()->first;
+    m_output_name = m_outputs.begin()->get_node()->get_name();
 }
 
 cv::GMat custom::BGMattingReplacer::replace(cv::GFrame in, cv::GMat bgr, const cv::Size& in_size, cv::GMat background) {