最近写了一段代码, 根据一个list找到对应的文件, 写入到tfrecord
中, 其中想要动态显示写入的进度, 于是去学习了一下发现sys.stdout
可以解决这个问题
def create_tf_record(output_filename, image_dir, label_dir, examples):
writer = tf.python_io.TFRecordWriter(output_filename)
for idx, example in enumerate(examples):
sys.stdout.write('\r>> Converting image %d/%d' % (idx + 1, len(examples)))
sys.stdout.flush()
if (idx+1) % 10000 == 0:
break
image_path = os.path.join(image_dir, example)
label_path = os.path.join(label_dir, example)
tf_example = dict_to_tf_example(image_path, label_path)
writer.write(tf_example.SerializeToString())
sys.stdout.write('\n')
sys.stdout.flush()
writer.close()
解释: sys.stdout.write()
向缓冲区中写入内容, sys.stdout.flush()
强制输出缓冲区的内容, 同时缓冲区被清空, '\r'
的意思是回车, '\n'
的意思是换行