update

MaybeShewill-CV · Nov 2, 2023 · 716fa94 · 716fa94
1 parent fc5ad2b
commit 716fa94
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 8 deletions.
diff --git a/src/models/classification/dinov2.inl b/src/models/classification/dinov2.inl
@@ -171,7 +171,7 @@ public:
 private:
     std::string _m_model_file_path;
     // mnn interpreter
-    std::unique_ptr <MNN::Interpreter> _m_net = nullptr;
+    MNN::Interpreter* _m_net = nullptr;
     // mnn session
     MNN::Session* _m_session = nullptr;
     // mnn session config
@@ -226,8 +226,7 @@ StatusCode Dinov2<INPUT, OUTPUT>::Impl::init(const decltype(toml::parse(""))& co
         return StatusCode::MODEL_INIT_FAILED;
     }
 
-    _m_net = std::unique_ptr<MNN::Interpreter>(MNN::Interpreter::createFromFile(_m_model_file_path.c_str()));
-
+    _m_net = MNN::Interpreter::createFromFile(_m_model_file_path.c_str());
     if (_m_net == nullptr) {
         LOG(ERROR) << "Create Interpreter failed, model file path: " << _m_model_file_path;
         _m_successfully_initialized = false;

diff --git a/src/models/classification/mobilenetv2.inl b/src/models/classification/mobilenetv2.inl
@@ -171,7 +171,7 @@ public:
 private:
     std::string _m_model_file_path;
     // MNN Net
-    std::unique_ptr <MNN::Interpreter> _m_net = nullptr;
+    MNN::Interpreter* _m_net = nullptr;
     // MNN Session
     MNN::Session* _m_session = nullptr;
     // MNN Session Config
@@ -226,7 +226,7 @@ StatusCode MobileNetv2<INPUT, OUTPUT>::Impl::init(const decltype(toml::parse("")
         return StatusCode::MODEL_INIT_FAILED;
     }
 
-    _m_net = std::unique_ptr<MNN::Interpreter>(MNN::Interpreter::createFromFile(_m_model_file_path.c_str()));
+    _m_net = MNN::Interpreter::createFromFile(_m_model_file_path.c_str());
     if (_m_net == nullptr) {
         LOG(ERROR) << "Create Interpreter failed, model file path: " << _m_model_file_path;
         _m_successfully_initialized = false;

diff --git a/src/models/classification/resnet.inl b/src/models/classification/resnet.inl
@@ -171,7 +171,7 @@ public:
 private:
     std::string _m_model_file_path;
     // MNN Net
-    std::unique_ptr <MNN::Interpreter> _m_net = nullptr;
+    MNN::Interpreter* _m_net = nullptr;
     // MNN session
     MNN::Session* _m_session = nullptr;
     // MNN session
@@ -226,8 +226,7 @@ StatusCode ResNet<INPUT, OUTPUT>::Impl::init(const decltype(toml::parse(""))& co
         return StatusCode::MODEL_INIT_FAILED;
     }
 
-    _m_net = std::unique_ptr<MNN::Interpreter>(MNN::Interpreter::createFromFile(_m_model_file_path.c_str()));
-
+    _m_net = MNN::Interpreter::createFromFile(_m_model_file_path.c_str());
     if (_m_net == nullptr) {
         LOG(ERROR) << "Create Interpreter failed, model file path: " << _m_model_file_path;
         _m_successfully_initialized = false;