未验证 提交 4a1a2d9d 编写于 作者: N niefeng 提交者: GitHub

fix mot infer image rgb (#8311)

上级 6c2ee01b
......@@ -375,8 +375,9 @@ class JDE_Detector(Detector):
if mot_skip_frame_num > 1 and frame_id > 0 and frame_id % mot_skip_frame_num > 0:
reuse_det_result = True
seq_name = video_out_name.split('.')[0]
frame_rgb = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
mot_results = self.predict_image(
[frame],
[frame_rgb],
visual=False,
seq_name=seq_name,
reuse_det_result=reuse_det_result)
......
......@@ -681,8 +681,9 @@ class SDE_Detector(Detector):
if mot_skip_frame_num > 1 and frame_id > 0 and frame_id % mot_skip_frame_num > 0:
reuse_det_result = True
seq_name = video_out_name.split('.')[0]
frame_rgb = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
mot_results = self.predict_image(
[frame],
[frame_rgb],
visual=False,
seq_name=seq_name,
reuse_det_result=reuse_det_result,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册