安装
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