未验证 提交 cb0dcc12 编写于 作者: H huangjun12 提交者: GitHub

fix bmn/bsn bug caused by the global variable in post processing (#4134)

上级 871f6162
......@@ -100,6 +100,7 @@ def soft_nms(df, alpha, t1, t2):
def video_process(video_list,
video_dict,
output_path,
result_dict,
snms_alpha=0.4,
snms_t1=0.55,
snms_t2=0.9):
......@@ -134,15 +135,13 @@ def bmn_post_processing(video_dict, subset, output_path, result_path):
num_videos_per_thread]
p = mp.Process(
target=video_process,
args=(
tmp_video_list,
video_dict,
output_path, ))
args=(tmp_video_list, video_dict, output_path, result_dict))
p.start()
processes.append(p)
tmp_video_list = video_list[(pp_num - 1) * num_videos_per_thread:]
p = mp.Process(
target=video_process, args=(tmp_video_list, video_dict, output_path))
target=video_process,
args=(tmp_video_list, video_dict, output_path, result_dict))
p.start()
processes.append(p)
for p in processes:
......
......@@ -104,6 +104,7 @@ def soft_nms(df, alpha, t1, t2):
def video_process(video_list,
video_dict,
output_path_pem,
result_dict,
snms_alpha=0.75,
snms_t1=0.65,
snms_t2=0.9):
......@@ -139,19 +140,13 @@ def bsn_post_processing(video_dict, subset, output_path_pem, result_path_pem):
num_videos_per_thread]
p = mp.Process(
target=video_process,
args=(
tmp_video_list,
video_dict,
output_path_pem, ))
args=(tmp_video_list, video_dict, output_path_pem, result_dict))
p.start()
processes.append(p)
tmp_video_list = video_list[(pp_num - 1) * num_videos_per_thread:]
p = mp.Process(
target=video_process,
args=(
tmp_video_list,
video_dict,
output_path_pem, ))
args=(tmp_video_list, video_dict, output_path_pem, result_dict))
p.start()
processes.append(p)
for p in processes:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册