From 00a2cb1c8646dfa8c6131f8d7bcefc46c2b92e9b Mon Sep 17 00:00:00 2001
From: cmhu <2657262686@qq.com>
Date: Mon, 31 Mar 2025 11:00:14 +0800
Subject: [PATCH] 修复vpt释放会报错的问题

---
 src/ai_engine_module/VPTProcess.cpp | 1 +
 src/ai_engine_module/VPTProcess.h   | 3 ---
 2 files changed, 1 insertion(+), 3 deletions(-)

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<DeviceMemory*> 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;
--
libgit2 0.21.4