diff --git a/deploy/pipeline/pipeline.py b/deploy/pipeline/pipeline.py index 5a578daa295..76aebe341ab 100644 --- a/deploy/pipeline/pipeline.py +++ b/deploy/pipeline/pipeline.py @@ -805,9 +805,9 @@ def predict_video(self, video_file, thread_idx=0): 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 08be7d2c619..53791517a86 100644 --- a/deploy/serving/python/web_service.py +++ b/deploy/serving/python/web_service.py @@ -62,7 +62,6 @@ def parse_args(self, argv=None): 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 @@ def get_model_vars(model_dir, service_config): 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)