python-3.x - 如何在 Tensorflow 2.0 中使用 tf.python_io.TFRecordWriter

Question

我想将 .csv 文件转换为 TF 记录。我现在的问题是，python_io 在 Tensorflow 2.0 中不存在。

     writer=tf.python_io.TFRecordWriter(FLAGS.output_path)
     path = os.path.join(FLAGS.image_dir)
     examples = pd.read_csv(FLAGS.csv_input)
     grouped = split(examples, 'filename')
     for group in grouped:
        tf_example = create_tf_example(group, path)
        writer.write(tf_example.SerializeToString())
     writer.close()
     output_path = os.path.join(os.getcwd(), FLAGS.output_path)

I get this error:
File "generate_tfrecord.py", line 102, in <module>
main()
File "generate_tfrecord.py", line 89, in main
writer = tf.python_io.TFRecordWriter(FLAGS.output_path)
AttributeError: module 'tensorflow' has no attribute 'python_io'

我必须改用什么？

Answer 1

据官方documentation ，python.io 包已移至名为 io 的新包。只需交换它们即可。

writer = tf.io.TFRecordWriter(FLAGS.output_path)
# ...