inference-engine/ie_bridges/python/sample/benchmark_app/benchmark_app.py

   1 import benchmark
   2
   3 from argparse import ArgumentParser, SUPPRESS
   4
   5
   6 def parse_args():
   7     parser = ArgumentParser(add_help=False)
   8     args = parser.add_argument_group('Options')
   9     args.add_argument('-h', '--help', action='help', default=SUPPRESS, help=benchmark.HELP_MESSAGES["HELP"])
  10     args.add_argument('-i', '--path_to_images', type=str, required=True,
  11                       help=benchmark.HELP_MESSAGES['IMAGE_MESSAGE'])
  12     args.add_argument('-m', '--path_to_model', type=str, required=True,
  13                       help=benchmark.HELP_MESSAGES['MODEL_MESSAGE'])
  14     args.add_argument('-c', '--path_to_cldnn_config', type=str, required=False,
  15                       help=benchmark.HELP_MESSAGES['CUSTOM_GPU_LIBRARY_MESSAGE'])
  16     args.add_argument('-l', '--path_to_extension', type=str, required=False, default=None,
  17                       help=benchmark.HELP_MESSAGES['CUSTOM_GPU_LIBRARY_MESSAGE'])
  18     args.add_argument('-api', '--api_type', type=str, required=False, default='async', choices=['sync', 'async'],
  19                       help=benchmark.HELP_MESSAGES['API_MESSAGE'])
  20     args.add_argument('-d', '--target_device', type=str, required=False, default="CPU",
  21                       help=benchmark.HELP_MESSAGES['TARGET_DEVICE_MESSAGE'])
  22     args.add_argument('-niter', '--number_iterations', type=int, required=False, default=None,
  23                       help=benchmark.HELP_MESSAGES['ITERATIONS_COUNT_MESSAGE'])
  24     args.add_argument('-nireq', '--number_infer_requests', type=int, required=False, default=2,
  25                       help=benchmark.HELP_MESSAGES['INFER_REQUESTS_COUNT_MESSAGE'])
  26     args.add_argument('-nthreads', '--number_threads', type=int, required=False, default=None,
  27                       help=benchmark.HELP_MESSAGES['INFER_NUM_THREADS_MESSAGE'])
  28     args.add_argument('-b', '--batch_size', type=int, required=False, default=None,
  29                       help=benchmark.HELP_MESSAGES['BATCH_SIZE_MESSAGE'])
  30     args.add_argument('-pin', '--infer_threads_pinning', type=str, required=False, default='YES',
  31                       choices=['YES', 'NO'], help=benchmark.HELP_MESSAGES['INFER_THREADS_PINNING_MESSAGE'])
  32     return parser.parse_args()
  33
  34
  35 if __name__ == "__main__":
  36     args = parse_args()
  37     benchmark.main(args)