优化显存占用,暂不缓存报警时刻的抓拍图

Zhao Shuaihua
1 parent b17e0e69
Showing 10 changed files with 26 additions and 6 deletions
src/ai_engine_module/motocycle_hs_process.cpp
src/ai_engine_module/motocycle_phone_process.cpp
src/ai_engine_module/motocycle_refit_phone_process.cpp
src/ai_engine_module/motocycle_refit_process.cpp
src/ai_engine_module/road_seg_correlation_algor.cpp
src/ai_engine_module/traffic_light_process.cpp
src/ai_engine_module/tricycle_manned_process.cpp
src/ai_engine_module/truck_manned_process.cpp
src/ai_platform/MultiSourceProcess.cpp
src/demo/demo.cpp
@@ -351,6 +351,7 @@ namespace ai_engine_module
                                     result.taskId = det_result.taskId;
                                     result.objId = det_result.objId;
                                     result.algor_type = algor_type; 
+#if 0 /*暂不保存报警时刻的抓拍图，有需要再启用*/
                                     // 原图
                                     vpc_img_info src_img_info = VPCUtil::vpc_devMem2vpcImg(src_img);
                                     result.origin_img_desc = src_img_info.pic_desc;
@@ -376,6 +377,7 @@ namespace ai_engine_module
                                     acldvppSetPicDescSize(vpcInputDesc_, nBufferSize);
                                     result.roi_img_desc = vpcInputDesc_; //需复制
+#endif
                                 }
                                 id_to_result_.emplace(obj_key, std::move(result));
                             }
@@ -346,6 +346,7 @@ namespace ai_engine_module
                                     result.taskId = det_result.taskId;
                                     result.objId = det_result.objId;
                                     result.algor_type = algor_type; 
+#if 0 /*暂不保存报警时刻的抓拍图，有需要再启用*/
                                     // 原图
                                     vpc_img_info src_img_info = VPCUtil::vpc_devMem2vpcImg(src_img);
                                     result.origin_img_desc = src_img_info.pic_desc;
@@ -371,6 +372,7 @@ namespace ai_engine_module
                                     acldvppSetPicDescSize(vpcInputDesc_, nBufferSize);
                                     result.roi_img_desc = vpcInputDesc_; //需复制
+#endif
                                 }
                                 id_to_result_.emplace(obj_key, std::move(result));
                             }
@@ -366,6 +366,7 @@ namespace ai_engine_module
                                     result.taskId = det_result.taskId;
                                     result.objId = det_result.objId;
                                     result.algor_type = algor_type; 
+#if 0 /*暂不保存报警时刻的抓拍图，有需要再启用*/
                                     // 原图
                                     vpc_img_info src_img_info = VPCUtil::vpc_devMem2vpcImg(src_img);
                                     result.origin_img_desc = src_img_info.pic_desc;
@@ -391,6 +392,7 @@ namespace ai_engine_module
                                     acldvppSetPicDescSize(vpcInputDesc_, nBufferSize);
                                     result.roi_img_desc = vpcInputDesc_; //需复制
+#endif
                                 }
                                 id_to_result_.emplace(obj_key, std::move(result));
                             }
@@ -342,6 +342,7 @@ namespace ai_engine_module
                                     result.taskId = det_result.taskId;
                                     result.objId = det_result.objId;
                                     result.algor_type = algor_type; 
+#if 0 /*暂不保存报警时刻的抓拍图，有需要再启用*/
                                     // 原图
                                     vpc_img_info src_img_info = VPCUtil::vpc_devMem2vpcImg(src_img);
                                     result.origin_img_desc = src_img_info.pic_desc;
@@ -367,6 +368,7 @@ namespace ai_engine_module
                                     acldvppSetPicDescSize(vpcInputDesc_, nBufferSize);
                                     result.roi_img_desc = vpcInputDesc_; //需复制
+#endif
                                 }
                                 id_to_result_.emplace(obj_key, std::move(result));
                             }
@@ -901,6 +901,7 @@ namespace ai_engine_module
                                     result.taskId = det_result.taskId;
                                     result.objId = det_result.objId;
                                     result.algor_type = algor_type; 
+#if 0 /*暂不保存报警时刻的抓拍图，有需要再启用*/
                                     // 原图
                                     vpc_img_info src_img_info = VPCUtil::vpc_devMem2vpcImg(src_img);
                                     result.origin_img_desc = src_img_info.pic_desc;
@@ -926,6 +927,7 @@ namespace ai_engine_module
                                     acldvppSetPicDescSize(vpcInputDesc_, nBufferSize);
                                     result.roi_img_desc = vpcInputDesc_; //需复制
+#endif
                                 }
                                 id_to_result_.emplace(obj_key, std::move(result));
                             }
@@ -611,6 +611,7 @@ namespace ai_engine_module
                                     result.taskId = det_result.taskId;
                                     result.objId = det_result.objId;
                                     result.algor_type = algor_type; 
+#if 0 /*暂不保存报警时刻的抓拍图，有需要再启用*/
                                     // 原图
                                     vpc_img_info src_img_info = VPCUtil::vpc_devMem2vpcImg(src_img);
                                     result.origin_img_desc = src_img_info.pic_desc;
@@ -636,6 +637,7 @@ namespace ai_engine_module
                                     acldvppSetPicDescSize(vpcInputDesc_, nBufferSize);
                                     result.roi_img_desc = vpcInputDesc_; //需复制
+#endif
                                 }
                                 id_to_result_.emplace(obj_key, std::move(result));
                             }
@@ -255,11 +255,15 @@ namespace ai_engine_module
                                 result.box = det_result.box;
                                 result.taskId = det_result.taskId;
                                 result.objId = det_result.objId;
+#if 0 /*暂不保存报警时刻的抓拍图，有需要再启用*/
                                 // 原图
                                 vpc_img_info src_img_info = VPCUtil::vpc_devMem2vpcImg(src_img);
                                 result.origin_img_desc = src_img_info.pic_desc;
                                 // 抠图
                                 result.roi_img_desc = flattened_vpc_imgs[n].pic_desc;
+#else
+                                VPCUtil::vpc_img_release(flattened_vpc_imgs[n]);
+#endif
                             }
                             id_to_result_.emplace(obj_key, std::move(result));
                             goto _continue;
@@ -478,11 +478,15 @@ namespace ai_engine_module
                                 result.box = det_result.box;
                                 result.taskId = det_result.taskId;
                                 result.objId = det_result.objId;
+#if 0 /*暂不保存报警时刻的抓拍图，有需要再启用*/
                                 // 原图
                                 vpc_img_info src_img_info = VPCUtil::vpc_devMem2vpcImg(src_img);
                                 result.origin_img_desc = src_img_info.pic_desc;
                                 // 抠图
                                 result.roi_img_desc = flattened_vpc_imgs[n].pic_desc;
+#else
+                                VPCUtil::vpc_img_release(flattened_vpc_imgs[n]);
+#endif
                             }
                             id_to_result_.emplace(obj_key, std::move(result));
                             goto _continue;
@@ -1727,7 +1727,7 @@ void CMultiSourceProcess::village_snapshot(vector&lt;string&gt;&amp; vpt_interest_task_id,
       std::string video_folder = "", result_folder = "", result_folder_little = ""; //要求各事件使用同一个视频保存地址，否则会被最后一个事件的地址覆盖
       auto task_other_params = m_task_param_manager->get_task_other_param(task_id);
       std::vector<int> algorithm_types;
-      bool save_single_algor_pic = false; // 是否保存单算法的报警图片
+      bool save_single_algor_pic = false; // 是否保存单算法的报警图片(若设置为true需先去掉算法内部注释!!!)
       if (task_param_ptr->nonmotor_vehicle_algors.find(algorithm_type_t::TRICYCLE_MANNED) != task_param_ptr->nonmotor_vehicle_algors.end()) {
         const auto &algor_other_params = task_other_params->find(algorithm_type_t::TRICYCLE_MANNED);
@@ -1155,11 +1155,11 @@ void test_gpu(int gpuID){
     switch (ch)
     {
     case 'a':
-      createTask(handle, algor_vec3, 4, false);  
-      createTask(handle, algor_vec3, 5, false);  
-      createTask(handle, algor_vec3, 6, false);  
-      createTask(handle, algor_vec3, 7, false);  
-      createTask(handle, algor_vec3, 8, false);  
+      createTask(handle, algor_vec2, 4, false);  
+      createTask(handle, algor_vec2, 5, false);  
+      createTask(handle, algor_vec2, 6, false);  
+      createTask(handle, algor_vec2, 7, false);  
+      createTask(handle, algor_vec2, 8, false);  
       break;
     case 'c':
       close_all_task(handle);