tools/generate_datafile/tf_dataset_converter/argparser.py

   1 '''Parse arguments'''
   2
   3 import argparse
   4
   5
   6 def _create_parser():
   7     parser = argparse.ArgumentParser(
   8         description='Convert a dataset of tensorflow to onert format')
   9     parser.add_argument(
  10         '-s', '--show-datasets', action='store_true', help='show dataset list')
  11     parser.add_argument(
  12         '-d',
  13         '--dataset-name',
  14         type=str,
  15         default='fashion_mnist',
  16         metavar='Dataset',
  17         help='name of dataset to be converted (default: "fashion_mnist")')
  18     parser.add_argument(
  19         '-o',
  20         '--out-dir',
  21         type=str,
  22         default='out',
  23         metavar='Dir',
  24         help='relative path of the files to be created (default: "out")')
  25     parser.add_argument(
  26         '-p',
  27         '--prefix-name',
  28         type=str,
  29         default='',
  30         metavar='Prefix',
  31         help='prefix name of the file to be created (default: "")')
  32     parser.add_argument(
  33         '-l',
  34         '--train-length',
  35         type=int,
  36         default=1000,
  37         metavar='N',
  38         help='Number of data for training (default: 1000)')
  39     parser.add_argument(
  40         '-t',
  41         '--test-length',
  42         type=int,
  43         default=100,
  44         metavar='N',
  45         help='Number of data for training (default: 100)')
  46
  47     return parser
  48
  49
  50 def parse_args():
  51     parser = _create_parser()
  52     args = parser.parse_args()
  53
  54     return args