From 843330075fa2d52f1f87f506d8bc7d7467744af3 Mon Sep 17 00:00:00 2001 From: zhiboniu <31800336+zhiboniu@users.noreply.github.com> Date: Tue, 14 Mar 2023 14:11:30 +0800 Subject: [PATCH] fix video_vis args and web service args (#7932) --- deploy/pipeline/pipeline.py | 6 +++--- deploy/serving/python/web_service.py | 2 +- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/deploy/pipeline/pipeline.py b/deploy/pipeline/pipeline.py index 5a578daa2..76aebe341 100644 --- a/deploy/pipeline/pipeline.py +++ b/deploy/pipeline/pipeline.py @@ -805,9 +805,9 @@ class PipePredictor(object): self.pipe_timer.total_time.end() if self.cfg['visual']: _, _, fps = self.pipe_timer.get_total_time() - im = self.visualize_video(frame_rgb, mot_res, frame_id, - fps, entrance, records, - center_traj) # visualize + im = self.visualize_video( + frame_rgb, mot_res, self.collector, frame_id, fps, + entrance, records, center_traj) # visualize if len(self.pushurl) > 0: pushstream.pipe.stdin.write(im.tobytes()) else: diff --git a/deploy/serving/python/web_service.py b/deploy/serving/python/web_service.py index 08be7d2c6..53791517a 100644 --- a/deploy/serving/python/web_service.py +++ b/deploy/serving/python/web_service.py @@ -62,7 +62,6 @@ class ArgsParser(ArgumentParser): assert args.config is not None, \ "Please specify --config=configure_file_path." args.service_config = self._parse_opt(args.opt, args.config) - print("args config:", args.service_config) args.model_config = PredictConfig(args.model_dir) return args @@ -254,6 +253,7 @@ if __name__ == '__main__': GLOBAL_VAR['fetch_vars'] = fetch_vars GLOBAL_VAR['preprocess_ops'] = FLAGS.model_config.preprocess_infos GLOBAL_VAR['model_config'] = FLAGS.model_config + print(FLAGS) # define the service uci_service = DetectorService(name="ppdet") uci_service.prepare_pipeline_config(yml_dict=FLAGS.service_config) -- GitLab