博客展示了一段Python代码，定义了total_sample函数，用于计算指定目录下TFRecord文件中的样本总数。通过遍历目录获取TFRecord文件路径，再使用tf_record_iterator迭代最后一个文件来统计样本数量，并给出了调用示例。
def total_sample(file_path):
tfrecords_list = []
temp = os.listdir(file_path)
# Get the path of tfrecord in a dir
sample_nums = 0
for tfrecord in temp:
tfrecords_list.append(os.path.join(file_path, tfrecord))
# print(tfrecords_list[-1])#返回list中的最后一个值
for record in tf.python_io.tf_record_iterator(tfrecords_list[-1]):
sample_nums += 1
# print(sample_nums)
return sample_nums
file_path = r’D:\python-soft\mtan\data\train’
i = total_sample(file_path)
扫一扫
1万+
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
