From 4a1a2d9d8f5aa5a64a5d86f92eba40818bc8bf05 Mon Sep 17 00:00:00 2001 From: niefeng Date: Tue, 13 Jun 2023 10:42:55 +0800 Subject: [PATCH] fix mot infer image rgb (#8311) --- deploy/pptracking/python/mot_jde_infer.py | 3 ++- deploy/pptracking/python/mot_sde_infer.py | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/deploy/pptracking/python/mot_jde_infer.py b/deploy/pptracking/python/mot_jde_infer.py index d381a211f..fb243d515 100644 --- a/deploy/pptracking/python/mot_jde_infer.py +++ b/deploy/pptracking/python/mot_jde_infer.py @@ -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) diff --git a/deploy/pptracking/python/mot_sde_infer.py b/deploy/pptracking/python/mot_sde_infer.py index 5a2693eec..b1c8cb3c9 100644 --- a/deploy/pptracking/python/mot_sde_infer.py +++ b/deploy/pptracking/python/mot_sde_infer.py @@ -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, -- GitLab