Merge pull request #20508 from TolyaTalamanov:at/expand-python-pyparams
authorAnatoliy Talamanov <anatoliy.talamanov@intel.com>
Fri, 6 Aug 2021 11:41:56 +0000 (14:41 +0300)
committerGitHub <noreply@github.com>
Fri, 6 Aug 2021 11:41:56 +0000 (14:41 +0300)
[G-API] Expand PyParams to support constInput

* Wrap constInputs to python

* Wrap cfgNumRequests

* Fix alignment

* Move macro to the line above

modules/gapi/include/opencv2/gapi/infer/bindings_ie.hpp
modules/gapi/src/backends/ie/bindings_ie.cpp

index fdd4128..92ef210 100644 (file)
@@ -22,17 +22,28 @@ namespace ie {
 // This class can be marked as SIMPLE, because it's implemented as pimpl
 class GAPI_EXPORTS_W_SIMPLE PyParams {
 public:
+    GAPI_WRAP
     PyParams() = default;
 
+    GAPI_WRAP
     PyParams(const std::string &tag,
              const std::string &model,
              const std::string &weights,
              const std::string &device);
 
+    GAPI_WRAP
     PyParams(const std::string &tag,
              const std::string &model,
              const std::string &device);
 
+    GAPI_WRAP
+    PyParams& constInput(const std::string &layer_name,
+                         const cv::Mat &data,
+                         TraitAs hint = TraitAs::TENSOR);
+
+    GAPI_WRAP
+    PyParams& cfgNumRequests(size_t nireq);
+
     GBackend      backend() const;
     std::string   tag()     const;
     cv::util::any params()  const;
index 35191d7..5874fe1 100644 (file)
@@ -37,3 +37,15 @@ cv::gapi::ie::PyParams cv::gapi::ie::params(const std::string &tag,
                                             const std::string &device) {
     return {tag, model, device};
 }
+
+cv::gapi::ie::PyParams& cv::gapi::ie::PyParams::constInput(const std::string &layer_name,
+                                                           const cv::Mat &data,
+                                                           TraitAs hint) {
+    m_priv->constInput(layer_name, data, hint);
+    return *this;
+}
+
+cv::gapi::ie::PyParams& cv::gapi::ie::PyParams::cfgNumRequests(size_t nireq) {
+    m_priv->cfgNumRequests(nireq);
+    return *this;
+}