python之absl的app,flags,loggig使用

安装

conda install absl-py

参数入口使用

from absl import app, flags, logging

flags.DEFINE_string('classes', './data/coco.names', 'path to classes file')
flags.DEFINE_boolean('tiny', False, 'yolov3 or yolov3-tiny')
flags.DEFINE_integer('size', 416, 'resize images to')
flags.DEFINE_float('gpu_fraction', 0.7, 'set gpu fraction')
flags.DEFINE_enum('mode', 'fit', ['fit', 'eager_fit', 'eager_tf'],
                  'fit: model.fit, '
                  'eager_fit: model.fit(run_eagerly=True), '
                  'eager_tf: custom GradientTape')

#...
FLAGS = flags.FLAGS

def main(_argv):
	print(FLAGS.classes)
	#...
	
if __name__ == '__main__':
    try:
        app.run(main)
    except SystemExit:
        pass