|
|
@@ -16,7 +16,8 @@
|
|
|
import numpy as np
|
|
|
|
|
|
|
|
|
-def human_seg_tracking(pre_gray, cur_gray, prev_cfd, dl_weights, disflow):
|
|
|
+def cal_optical_flow_tracking(pre_gray, cur_gray, prev_cfd, dl_weights,
|
|
|
+ disflow):
|
|
|
"""计算光流跟踪匹配点和光流图
|
|
|
输入参数:
|
|
|
pre_gray: 上一帧灰度图
|
|
|
@@ -59,7 +60,7 @@ def human_seg_tracking(pre_gray, cur_gray, prev_cfd, dl_weights, disflow):
|
|
|
return track_cfd, is_track, dl_weights
|
|
|
|
|
|
|
|
|
-def human_seg_track_fuse(track_cfd, dl_cfd, dl_weights, is_track):
|
|
|
+def fuse_optical_flow_tracking(track_cfd, dl_cfd, dl_weights, is_track):
|
|
|
"""光流追踪图和人像分割结构融合
|
|
|
输入参数:
|
|
|
track_cfd: 光流追踪图
|
|
|
@@ -116,9 +117,9 @@ def postprocess(cur_gray, scoremap, prev_gray, pre_cfd, disflow, is_init):
|
|
|
fusion_cfd = cur_cfd
|
|
|
else:
|
|
|
weights = np.ones((h, w), np.float32) * 0.3
|
|
|
- track_cfd, is_track, weights = human_seg_tracking(
|
|
|
+ track_cfd, is_track, weights = cal_optical_flow_tracking(
|
|
|
prev_gray, cur_gray, pre_cfd, weights, disflow)
|
|
|
- fusion_cfd = human_seg_track_fuse(track_cfd, cur_cfd, weights,
|
|
|
- is_track)
|
|
|
+ fusion_cfd = fuse_optical_flow_tracking(track_cfd, cur_cfd, weights,
|
|
|
+ is_track)
|
|
|
|
|
|
return fusion_cfd
|