命令行参数 ——tf.app.flags和python argparse

命令行参数 ——tf.app.flags和python argparse

• 深度学习神经网络往往有过多的Hyperparameter需要调优，优化算法、学习率、卷积核尺寸等很多参数都需要不断调整，使用命令行参数是非常方便的。有两种实现方式，一是利用python的argparse包，二是调用tensorflow自带的app.flags实现。

• tf.app.flags

import tensorflow as tf
flags = tf.app.flags
flags.DEFINE_string('path',
                        'data/test_data',
                        'Path to annotation')
 FLAGS = flags.FLAGS

命令行： python test.py --path 'data'

• python argparse

import argparse
parser = argparse.ArgumentParser(description='')
parser.add_argument('--path', default='data/test_data')

args = parser.parse_args()

命令行： python test.py --path 'data'