一次读取tfrecod文件与tfrecord中的样本数量

当使用tensorflow训练时候,由于生成tfrecord为多个文件,该怎么办那?可以参考如下代码:

tfrecord_file_path = '/train/*.tfrecords’#train是存放tfrecord的文件夹
filename_queue = tf.train.string_input_producer(
                              tf.train.match_filenames_once(tfrecord_file_path),
                              shuffle=True, num_epochs=None) #None表示没哟限制
reader = tf.TFRecordReader()
_, serialized_example = reader.read(filename_queue)   #返回文件名和文件
features = tf.parse_single_example(serialized_example,
                                       features={XXXXXXX})  #取出XXXXXXX

 

在用tensorflow进行模型训练时,一般都会保存为tfrecord格式的数据，有时候会想查看该tfrecord格式的数据样本多少个。该怎么办那?可以参考如下代码：

    import tensorflow as tf
     
    tf_records_filenames = 'xxx.tfrecords'
    c = 0
    for record in tf.python_io.tf_record_iterator(fn):
       c += 1 
    print c

参考:

1. https://blog.csdn.net/CSDNbaiduyun/article/details/81537837
2. https://blog.csdn.net/chanbo8205/article/details/85703719
3. https://blog.csdn.net/b876144622/article/details/79962786
4. https://blog.csdn.net/weixin_42499236/article/details/83998139
5. https://blog.csdn.net/chaipp0607/article/details/72960028