google absl.flags库替换操作实践教程

Posted 2020-11-25 xiexiaokui

google absl.flags库替换操作实践教程

绪论：

google在object detection（包括tensorflow和其他模块）中大量absl.flags。用此方法，可以不用修改源代码，即可直接方便的读取命令行参数，相当于py进行了一定程度的定制。

但此方法有两个显著的缺点：

1. 不兼容windows。测试中表明，路径分隔符 / 和 \ 可能混用，而且不能正确解析，导致文件无法打开。

2. 无调试功能，更无断点。只能通过大量的print或者log来反馈信息，极为低效。

思路：改为标准的py，而并非命令行调用。

步骤：

1.注释

#from absl.flags import FLAGS

2. 定义类

class FLAGS:

pass

3. 可通过以下命令查看当前工作路径：

import os

dir=os.getcwd()

print("dir",dir)

测试表明，在vs中直接运行py文件是，当前工作路径是项目文件夹。

4. 注释掉flags.DEFINE_

例如：

#flags.DEFINE_string(‘data_dir‘, ‘./data/voc2012_raw/VOCdevkit/VOC2012/‘,

# ‘path to raw PASCAL VOC dataset‘)

#flags.DEFINE_enum(‘split‘, ‘train‘, [

# ‘train‘, ‘val‘], ‘specify train or val spit‘)

#flags.DEFINE_string(‘output_file‘, ‘./data/voc2012_train.tfrecord‘, ‘outpot dataset‘)

#flags.DEFINE_string(‘classes‘, ‘./data/voc2012.names‘, ‘classes file‘)

5. 定义FLAGS成员

FLAGS.data_dir="./data/voc2012_raw/VOCdevkit/VOC2012"

FLAGS.split="train"

FLAGS.output_file="./data/voc2012_train.tfrecord"

FLAGS.classes="./data/voc2012.names"