优化结果文件夹配置

Hu Chunming
1 parent 603874de
Showing 3 changed files with 10 additions and 2 deletions
src/ai_platform/header.h
src/ai_platform/mvpt.cpp
src/ai_platform/task_manager.cpp
@@ -126,6 +126,9 @@ typedef struct task_param {
     const char *task_id;                      //外部传入任务id
     int dec_type;
     int skip_frame{0};
+
+    string result_folder_little;         //目标快照抠图保存地址
+	string result_folder;                //目标快照大图保存地址
 } task_param;
 #endif
  
@@ -542,9 +542,11 @@ void CMultiSourceProcess::vehicle_locus_finished(const OBJ_KEY obj_key) {
   }
  
   const OBJ_VALUE obj_value = it->second;
+
+  TaskInfo* info = m_task_manager.GetTaskInfo(obj_key.video_id);
  
-  string result_folder = "./res/snapshot";//todo
-  string result_folder_little = "./res/snapshot_lite";
+  string result_folder = info->result_folder + "/snapshot";//todo
+  string result_folder_little = info->result_folder_little + "/snapshot_lite";
  
   std::string cur_timestamp_ms =  std::to_string(helpers::timer::get_cur_time_ms());
   // 原图
@@ -25,6 +25,9 @@ bool task_manager::AddTask(const task_param&amp; task_param, FFDecConfig&amp; cfg) {
     info.ipc_url = task_param.ipc_url;
     info.task_id = task_param.task_id;
  
+    info.result_folder = task_param.result_folder;
+    info.result_folder_little = task_param.result_folder_little;
+
     DvppDecoder* dec = create_decoder(cfg);
     if(dec == nullptr) {
         return false;