def main():
##
# Handle command line input
##
num_clips = 50000
# set train frame dimensions
assert os.path.exists(c.TRAIN_DIR)
c.FULL_HEIGHT, c.FULL_WIDTH = c.get_train_frame_dims()
##
# Process data for training
##
process_training_data(num_clips)
def main():
##
# Handle command line input
##
num_clips = 5000000
try:
opts, _ = getopt.getopt(sys.argv[1:], 'w:n:t:c:oH', [
'num_workers=', 'num_clips=', 'train_dir=', 'clips_dir=',
'overwrite', 'help'
])
except getopt.GetoptError:
usage()
sys.exit(2)
for opt, arg in opts:
if opt in ('-w', '--num_workers'):
c.NUM_WORKERS = int(arg)
if opt in ('-n', '--num_clips'):
num_clips = int(arg)
if opt in ('-t', '--train_dir'):
c.TRAIN_DIR = c.get_dir(arg)
if opt in ('-c', '--clips_dir'):
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)
# set train frame dimensions
assert os.path.exists(c.TRAIN_DIR)
c.FULL_HEIGHT, c.FULL_WIDTH = c.get_train_frame_dims()
##
# Process data for training
##
process_training_data(num_clips)