diff options
Diffstat (limited to 'Code/process_data.py')
| -rw-r--r-- | Code/process_data.py | 12 |
1 files changed, 8 insertions, 4 deletions
diff --git a/Code/process_data.py b/Code/process_data.py index 170959a..446de91 100644 --- a/Code/process_data.py +++ b/Code/process_data.py @@ -28,12 +28,13 @@ def process_training_data(num_clips): def usage(): print 'Options:' - print '-n/--num_clips= <# clips to process for training>' + print '-n/--num_clips= <# clips to process for training> (Default = 5000000)' print '-t/--train_dir= <Directory of full training frames>' print '-c/--clips_dir= <Save directory for processed clips>' print " (I suggest making this a hidden dir so the filesystem doesn't freeze" print " with so many files. DON'T `ls` THIS DIRECTORY!)" - print '-o/--overwrite (Overwrites the previous data in the training dir)' + print '-o/--overwrite (Overwrites the previous data in clips_dir)' + print '-H/--help (Prints usage)' def main(): @@ -44,8 +45,8 @@ def main(): num_clips = 5000000 try: - opts, _ = getopt.getopt(sys.argv[1:], 'n:t:c:o', - ['num_clips=', 'train_dir=', 'clips_dir=', 'overwrite']) + opts, _ = getopt.getopt(sys.argv[1:], 'n:t:c:oH', + ['num_clips=', 'train_dir=', 'clips_dir=', 'overwrite', 'help']) except getopt.GetoptError: usage() sys.exit(2) @@ -59,6 +60,9 @@ def main(): c.TRAIN_DIR_CLIPS = c.get_dir(arg) if opt in ('-o', '--overwrite'): c.clear_dir(c.TRAIN_DIR_CLIPS) + if opt in ('-H', '--help'): + usage() + sys.exit(2) ## # Process data for training |