EADST

TensorFlow: Check the Number of Files in TFrecords

After packaging a tfrecord file, we need to check whether all the files are in the tfrecord file or not. Here is the code.

import tensorflow as tf

name = '.../test.tfrecords'
count = 0
for record in tf.python_io.tf_record_iterator(name):
    count += 1 
print(c)
相关标签
About Me
XD
Goals determine what you are going to be.
Category
标签云
站点统计

本站现有博文266篇,共被浏览440565

本站已经建立2019天!

热门文章
文章归档
回到顶部