未验证 提交 84333007 编写于 作者: Z zhiboniu 提交者: GitHub

fix video_vis args and web service args (#7932)

上级 9d8125d6
......@@ -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:
......
......@@ -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)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册