我正在使用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”
怎么了?
答案 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 ,以便针对抛出错误的这些特定方法修改您的代码。