我正在关注this tutorial并使用tensorflow进行自定义对象检测项目。
所以当我尝试使用以下命令为火车图像创建TF记录
python3 generate_tfrecord.py --csv_input=data/train_labels.csv --output_path=data/train.record
我收到以下错误:
Traceback (most recent call last):
File "generate_tfrecord.py", line 23, in <module>
flags = tf.app.flags
AttributeError: module 'tensorflow' has no attribute 'app'
如何解决此错误?
答案 0 :(得分:13)
尝试使用import tensorflow.compat.v1 as tf
答案 1 :(得分:7)
您使用的是哪个Tensorflow版本?如果是TF2.0,则由于不再支持tf.app.flags
defined here,因此需要将tf.compat.v1.flags
替换为build
docs。
答案 2 :(得分:0)
我也遇到了类似的错误,然后安装了 tensorflow 1.15 版本并且它在以下警告中起作用。
警告:tensorflow:来自 generate_tfrecord.py:104:名称 tf.app.run 已弃用。请改用 tf.compat.v1.app.run。
因此,如果您遇到此错误,请尝试使用 tf.compat.v1.app.run 而不是 tf.app.run,如 tensorflow 警告中所述。
答案 3 :(得分:0)
如果您不想降级 tf,请使用 absl。
from absl import app
if __name__ == '__main__':
app.run(main)