Commit 242dc71c authored by Yuxin Wu's avatar Yuxin Wu

update docs

parent 1cdd2e9e
...@@ -115,8 +115,8 @@ if __name__ == '__main__': ...@@ -115,8 +115,8 @@ if __name__ == '__main__':
parser.add_argument('--benchmark', action='store_true', help="Benchmark the speed of the model + postprocessing") parser.add_argument('--benchmark', action='store_true', help="Benchmark the speed of the model + postprocessing")
parser.add_argument('--config', help="A list of KEY=VALUE to overwrite those defined in config.py", parser.add_argument('--config', help="A list of KEY=VALUE to overwrite those defined in config.py",
nargs='+') nargs='+')
parser.add_argument('--compact', help='if you want to save a model to .pb') parser.add_argument('--compact', help='Save a model to .pb')
parser.add_argument('--serving', help='if you want to save a model to serving file') parser.add_argument('--serving', help='Save a model to serving file')
args = parser.parse_args() args = parser.parse_args()
if args.config: if args.config:
...@@ -147,7 +147,7 @@ if __name__ == '__main__': ...@@ -147,7 +147,7 @@ if __name__ == '__main__':
ModelExporter(predcfg).export_compact(args.compact, optimize=False) ModelExporter(predcfg).export_compact(args.compact, optimize=False)
elif args.serving: elif args.serving:
ModelExporter(predcfg).export_serving(args.serving, optimize=False) ModelExporter(predcfg).export_serving(args.serving, optimize=False)
if args.predict: if args.predict:
predictor = OfflinePredictor(predcfg) predictor = OfflinePredictor(predcfg)
for image_file in args.predict: for image_file in args.predict:
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment