tensorflow中的dataset

最新推荐文章于 2024-03-21 09:54:05 发布

洛北辰南

最新推荐文章于 2024-03-21 09:54:05 发布

阅读量854

点赞数

分类专栏：深度学习深度学习框架文章标签： tensorflow dataset

本文链接：https://blog.csdn.net/qq_18941713/article/details/93972874

版权

博客探讨了在TensorFlow中使用Dataset时，shuffle、batch和repeat方法的顺序对数据处理的影响。原始代码中，数据未被打乱，原因是shuffle发生在repeat之后。作者调整了顺序，先设定buffer_size打乱数据，再进行batch处理和repeat，确保数据顺序被正确地随机化，以优化训练效果。

摘要由CSDN通过智能技术生成

先记录一下读tfrecord的方式

def parser(record, shape=[224, 224, 1]):
    features = tf.parse_single_example(record, features={
   
        'label': tf.FixedLenFeature([], tf.int64),
        'img': tf.FixedLenFe