python tensorflow 命令行输入

tensorflow 命令行输入有两种：

1. 利用python自带库argparse

import argparse
import sys
def get_arguments():
    parser = argparse.ArgumentParser(description='example network')
    parser.add_argument('--batch_size', type=int,default=BATCH_SIZE,
                        help='How many files to process at once. Default:'+str(BATCH_SIZE)+'.')
    return parser.parse_args()

args = get_arguments()
batch_size = args.batch_size

2. 利用tensorflow的tf.flags组件

FLAGS = tf.flags.FLAGS

tf.flags.DEFINE_integer('batch_size', 1, 'batch size, default: 1')

batch_size = FLAGS.batch_size