Merge pull request #135 from Corsair-cxs/iss135

Comment function 'createLandmarksDetection' related code [issue #134]
intel · Mar 15, 2021 · c4b56ba · c4b56ba
2 parents 955b049 + 5d71caa
commit c4b56ba
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 6 deletions.
diff --git a/dynamic_vino_lib/include/dynamic_vino_lib/pipeline_manager.h b/dynamic_vino_lib/include/dynamic_vino_lib/pipeline_manager.h
@@ -117,8 +117,8 @@ class PipelineManager
   createVehicleAttribsDetection(const Params::ParamManager::InferenceRawData& infer);
   std::shared_ptr<dynamic_vino_lib::BaseInference>
   createLicensePlateDetection(const Params::ParamManager::InferenceRawData& infer);
-  std::shared_ptr<dynamic_vino_lib::BaseInference>
-  createLandmarksDetection(const Params::ParamManager::InferenceRawData& infer);
+  // std::shared_ptr<dynamic_vino_lib::BaseInference>
+  // createLandmarksDetection(const Params::ParamManager::InferenceRawData& infer);
 
   std::map<std::string, PipelineData> pipelines_;
   Engines::EngineManager engine_manager_;

diff --git a/dynamic_vino_lib/src/pipeline_manager.cpp b/dynamic_vino_lib/src/pipeline_manager.cpp
@@ -258,10 +258,10 @@ PipelineManager::parseInference(const Params::ParamManager::PipelineRawData& par
     {
       object = createPersonAttribsDetection(infer);
     }
-    else if (infer.name == kInferTpye_LandmarksDetection)
-    {
-      object = createLandmarksDetection(infer);
-    }
+    // else if (infer.name == kInferTpye_LandmarksDetection)
+    // {
+    //   object = createLandmarksDetection(infer);
+    // }
     else if (infer.name == kInferTpye_VehicleAttribsDetection)
     {
       object = createVehicleAttribsDetection(infer);
@@ -431,6 +431,20 @@ PipelineManager::createPersonAttribsDetection(const Params::ParamManager::Infere
   return attribs_inference_ptr;
 }
 
+// std::shared_ptr<dynamic_vino_lib::BaseInference>
+// PipelineManager::createLandmarksDetection(
+//   const Params::ParamManager::InferenceRawData & infer)
+// {
+//   auto model = std::make_shared<Models::LandmarksDetectionModel>(infer.model, infer.batch);
+//   model->modelInit();
+//   auto engine = engine_manager_.createEngine(infer.engine, model);
+//   auto landmarks_inference_ptr =  std::make_shared<dynamic_vino_lib::LandmarksDetection>();
+//   landmarks_inference_ptr->loadNetwork(model);
+//   landmarks_inference_ptr->loadEngine(engine);
+
+//   return landmarks_inference_ptr;
+// }
+
 void PipelineManager::threadPipeline(const char* name)
 {
   PipelineData& p = pipelines_[name];