python 蟒蛇参数解析

Posted 2021-05-09

from absl import flags
#或者直接用
#import tensorflow as tf
#flags = tf.app.flags

#参数：参数名，默认值，描述
flags.DEFINE_string(
    'model_dir', None, 'Path to output model directory '
    'where event and checkpoint files will be written.')
flags.DEFINE_integer('num_train_steps', None, 'Number of train steps.')
flags.DEFINE_boolean(
    'run_once', False, 'If running in eval-only mode, whether to run just '
    'one round of eval vs running continuously (default).'
)
FLAGS = flags.FLAGS

#调用方法 FLAGS.run_once

def parse_args():
    parser = argparse.ArgumentParser(description="Train CornerNet")
    parser.add_argument("cfg_file", help="config file", type=str) #不指定参数名直接传入参数
    parser.add_argument("--iter", dest="start_iter",help="train at iteration i",default=0, type=int)
    parser.add_argument("--flag", dest="flag", default=True, type=bool)#参数名，描述，默认，参数类型
    args = parser.parse_args()
    return args

#调用方法 args = parse_args(); args.flag