diff --git a/src/ai_platform/header.h b/src/ai_platform/header.h index f6fbd72..ce43aee 100755 --- a/src/ai_platform/header.h +++ b/src/ai_platform/header.h @@ -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 diff --git a/src/ai_platform/mvpt.cpp b/src/ai_platform/mvpt.cpp index 3c59058..a9da98f 100755 --- a/src/ai_platform/mvpt.cpp +++ b/src/ai_platform/mvpt.cpp @@ -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()); // 原图 diff --git a/src/ai_platform/task_manager.cpp b/src/ai_platform/task_manager.cpp index fc6e3f9..582414f 100644 --- a/src/ai_platform/task_manager.cpp +++ b/src/ai_platform/task_manager.cpp @@ -25,6 +25,9 @@ bool task_manager::AddTask(const task_param& task_param, FFDecConfig& 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;