我目前正在第 4 节学习本教程。当我运行命令以生成 TF 记录时,它会返回generate_tfrecord.py文件的回溯错误。第一个错误是:
flags = tf.compat.v1.flags
flags.DEFINE_string('csv_input', '', 'Path to the CSV input')
flags.DEFINE_string('image_dir', '', 'Path to the image directory')
flags.DEFINE_string('output_path', '', 'Path to output TFRecord')
FLAGS = flags.FLAGS
我只是通过添加 .compat.v1 行来修复它,因为我使用的是 TF 2.0。
我得到的下一个错误是最后一行;
if __name__ == '__main__':
tf.app.run()
它返回:
Traceback (most recent call last):
File "generate_tfrecord.py", line 101, in <module>
tf.app.run()
AttributeError: module 'tensorflow' has no attribute 'app'
任何帮助将不胜感激! -干杯
https://www.tensorflow.org/guide/effective_tf2,在这个TensorFlow 2指南中,它说tf.app
被删除了。 要解决此问题,请卸载 TensorFlow 2.x,然后安装 1.x,或者修改代码使其使用 2.x API。 你应该能够只调用main方法而不是tf.app.run((。
参考: https://github.com/tensorflow/tensorflow/issues/34431
或者你可以简单地添加import tensorflow.compat.v1 as tf
tf.disable_v2_behavior()
而不是import tensorflow as tf