fix video_vis args and web service args (#7931)

PaddlePaddle · Mar 13, 2023 · 08a0d75 · 08a0d75
1 parent 5dc90ee
commit 08a0d75
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git 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
@@ -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)