自定义对象检测AttributeError:模块'tensorflow'没有属性'app'

时间:2019-10-12 05:10:41

标签: python-3.x tensorflow object-detection

我正在使用tensorflow进行this project自定义对象检测。

当我尝试使用以下命令为火车图像创建TF记录时:

python generate_tfrecord.py 
  --csv_input=images\train_labels.csv 
  --image_dir=images\train 
  --output_path=train.record

我收到以下错误:

  

“ generate_tfrecord.py”,第102行,位于tf.app.run()
  AttributeError:模块“ tensorflow”没有属性“ app”

怎么了?

1 个答案:

答案 0 :(得分:0)

我遇到了同样的问题

您可以通过在generate_tfrecords.py文件中进行更改来解决相同问题

第45行:将tf.gfile.GFile(os.path.join(path,'{}'。format(group.filename)),'rb')替换为fid: 如 tftf.compat.v1.gfile.GFile(os.path.join(path,'{}'。format(group.filename)),'rb')as fid:

第23行:标志= tf.app.flags为 tf.compat.v1.flags 第86行:writer = writer = tf.python_io.TFRecordWriter(FLAGS.output_path)as t f.compat.v1.python_io.TFRecordWriter(FLAGS.output_path) 和 第100行:tf.app.run()为 tf.compat.v1.app.run()

基本上,某些方法是根据Tensorflow版本1.0构建的,因此对于这些方法,您可以使用 tf.compat.v1 ,以便针对抛出错误的这些特定方法修改您的代码。