tensorflow中read_up_to函数读多个文件时一次能读出来的记录数
采用如下代码从多个文件中读取数据时,read_up_to函数每次随机从文件名队列中抽取一个未读完的文件,从该文件中读取至多num_records条记录。若该文件中剩余未读记录数不足num_records条,则有几条就读几条出来。如果多次调用read_up_to函数,则优先读完上次没有读完的文件。之后再从剩下没读过的文件中随机抽取一个进行读取。如果所有文件都读完后继续调用read_up_to函数,则重新开始对整个数据集的新一轮读取。reader = tf.TFRecordReader()filename
原创
2021-05-15 16:20:38 ·
832 阅读 ·
0 评论