diff --git a/deploy/pipeline/pipeline.py b/deploy/pipeline/pipeline.py index 5a578daa29555395eb906c67701ac7df9f8e91e4..76aebe341abc572eb1a78a71a832268d7d8a8415 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 08be7d2c61904b3d3d2e21643c7eaddd3581f48b..53791517a86cf245cb269b1907459cce5c5ad471 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)