tensorflow统计tfrecord中的样本数量
·
在用tf处理数据时，一般都会保存为tfrecord格式的数据，有时候会想查看该tfrecord格式的数据样本多少个。那么如何做呢？
参考如下代码：
最后的输出c就是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最后的输出c就是tfrecord中的样本数量
AtomGit 是由开放原子开源基金会联合 CSDN 等生态伙伴共同推出的新一代开源与人工智能协作平台。平台坚持“开放、中立、公益”的理念，把代码托管、模型共享、数据集托管、智能体开发体验和算力服务整合在一起，为开发者提供从开发、训练到部署的一站式体验。
更多推荐
2
0- 0
已为社区贡献10条内容
所有评论(0)