我想在Jupyter Notebook中而不是从终端运行TensorFlow Object Detection API。特别是,我想通过调用具有train.py
功能的main
来开始培训
def main(_):
pass
在使用python train.py <additional flags>
运行文件时被调用,原因是文件中的这一部分:
if __name__ == '__main__':
tf.app.run()
我发现它在从终端中传递的标志中设置了全局变量后,调用了main函数以开始运行:
flags = tf.app.flags
flags.DEFINE_string('master', '', 'Name of the TensorFlow master to use.')
...
有没有一种方法可以设置全局变量flags
,而无需更改脚本train.py
,然后通过导入文件来调用main函数。不幸的是,我在tf.app.flags
上找不到任何文档。