From 5d71caa54c933a708e4881d216d54ff8daf81c2c Mon Sep 17 00:00:00 2001 From: Corsair-cxs <1494089474@qq.com> Date: Sun, 14 Mar 2021 17:59:26 +0800 Subject: [PATCH] Comment function 'createLandmarksDetection' related code --- .../dynamic_vino_lib/pipeline_manager.h | 4 ++-- dynamic_vino_lib/src/pipeline_manager.cpp | 22 +++++++++++++++---- 2 files changed, 20 insertions(+), 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 index d131c530..f0cb62f3 100644 --- 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 createLicensePlateDetection(const Params::ParamManager::InferenceRawData& infer); - std::shared_ptr - createLandmarksDetection(const Params::ParamManager::InferenceRawData& infer); + // std::shared_ptr + // createLandmarksDetection(const Params::ParamManager::InferenceRawData& infer); std::map pipelines_; Engines::EngineManager engine_manager_; diff --git a/dynamic_vino_lib/src/pipeline_manager.cpp b/dynamic_vino_lib/src/pipeline_manager.cpp index 7ad6b54b..6e43dbd3 100644 --- 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 +// PipelineManager::createLandmarksDetection( +// const Params::ParamManager::InferenceRawData & infer) +// { +// auto model = std::make_shared(infer.model, infer.batch); +// model->modelInit(); +// auto engine = engine_manager_.createEngine(infer.engine, model); +// auto landmarks_inference_ptr = std::make_shared(); +// landmarks_inference_ptr->loadNetwork(model); +// landmarks_inference_ptr->loadEngine(engine); + +// return landmarks_inference_ptr; +// } + void PipelineManager::threadPipeline(const char* name) { PipelineData& p = pipelines_[name];