一般需要用来统计样本数量的:
# 该函数用于统计 TFRecord 文件中的样本数量(总数)
def total_sample(file_name):
sample_nums = 0
for record in tf.python_io.tf_record_iterator(file_name):
sample_nums += 1
return sample_nums
一般需要用来统计样本数量的:
# 该函数用于统计 TFRecord 文件中的样本数量(总数)
def total_sample(file_name):
sample_nums = 0
for record in tf.python_io.tf_record_iterator(file_name):
sample_nums += 1
return sample_nums