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