diff --git a/src/ai_engine_module/VPTProcess.cpp b/src/ai_engine_module/VPTProcess.cpp index 8531a89..236d3c3 100755 --- a/src/ai_engine_module/VPTProcess.cpp +++ b/src/ai_engine_module/VPTProcess.cpp @@ -191,6 +191,7 @@ int VPTProcess::process(vector vec_gpuMem, /* 算法句柄 资源释放 */ void VPTProcess::release(){ if (m_det_handle){ + aclrtSetCurrentContext(m_algorthim_ctx); vpt_release(&m_det_handle); m_det_handle = NULL; } diff --git a/src/ai_engine_module/VPTProcess.h b/src/ai_engine_module/VPTProcess.h index 3755a7b..23e8706 100755 --- a/src/ai_engine_module/VPTProcess.h +++ b/src/ai_engine_module/VPTProcess.h @@ -13,8 +13,6 @@ #include "sort/Sort.h" #include "acl/acl.h" #include "acl/ops/acl_dvpp.h" -#include "../util/vpc_util.h" -#include "../util/JpegUtil.h" #include "ai_engine_header.h" @@ -78,7 +76,6 @@ public: private: int m_devId; aclrtContext m_algorthim_ctx; - JpegUtil jpegUtil; void* m_det_handle{nullptr}; float threshold{0.6}; int m_max_batchsize;