修复vpt释放会报错的问题

Hu Chunming
1 parent 1fa39585
Showing 2 changed files with 1 additions and 3 deletions
src/ai_engine_module/VPTProcess.cpp
src/ai_engine_module/VPTProcess.h
@@ -191,6 +191,7 @@ int VPTProcess::process(vector&lt;DeviceMemory*&gt; vec_gpuMem,
 /* 算法句柄 资源释放 */
 void VPTProcess::release(){
     if (m_det_handle){
+        aclrtSetCurrentContext(m_algorthim_ctx);
         vpt_release(&m_det_handle);
         m_det_handle = NULL;
     }
@@ -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;