Skip to content

P
PaddleDetection

PaddlePaddle / PaddleDetection
大约 1 年 前同步成功

通知 695
Star 11112
Fork 2696
P
PaddleDetection
未验证 提交 4a1a2d9d 编写于 作者: N niefeng 提交者： GitHub
浏览文件
操作

fix mot infer image rgb (#8311)

上级 6c2ee01b
显示空白变更内容
内联 并排
Showing with 4 addition and 2 deletion
deploy/pptracking/python/mot_jde_infer.py
浏览文件 @ 4a1a2d9d
...@@ -375,8 +375,9 @@ class JDE_Detector(Detector): ...@@ -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: if mot_skip_frame_num > 1 and frame_id > 0 and frame_id % mot_skip_frame_num > 0:
reuse_det_result = True reuse_det_result = True
seq_name = video_out_name.split('.')[0] seq_name = video_out_name.split('.')[0]
frame_rgb = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
mot_results = self.predict_image( mot_results = self.predict_image(
[frame], [frame_rgb],
visual=False, visual=False,
seq_name=seq_name, seq_name=seq_name,
reuse_det_result=reuse_det_result) reuse_det_result=reuse_det_result)
......
deploy/pptracking/python/mot_sde_infer.py
浏览文件 @ 4a1a2d9d
...@@ -681,8 +681,9 @@ class SDE_Detector(Detector): ...@@ -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: if mot_skip_frame_num > 1 and frame_id > 0 and frame_id % mot_skip_frame_num > 0:
reuse_det_result = True reuse_det_result = True
seq_name = video_out_name.split('.')[0] seq_name = video_out_name.split('.')[0]
frame_rgb = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
mot_results = self.predict_image( mot_results = self.predict_image(
[frame], [frame_rgb],
visual=False, visual=False,
seq_name=seq_name, seq_name=seq_name,
reuse_det_result=reuse_det_result, reuse_det_result=reuse_det_result,
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或