DataReaders.py 文件源码

python
阅读 24 收藏 0 点赞 0 评论 0

项目:TFExperiments 作者: gnperdue 项目源码 文件源码
def _tfrecord_to_graph_ops(self, num_epochs):
        with tf.variable_scope('tfrec_to_graph'):
            file_queue = tf.train.string_input_producer(
                self.filenames_list,
                name=self.name+'_file_queue',
                num_epochs=num_epochs
            )
            reader = tf.TFRecordReader(
                options=tf.python_io.TFRecordOptions(
                    compression_type=self.compression
                ), name=self.name+'_tfrec_reader'
            )
            _, tfrecord = reader.read(file_queue)

            features, targets = parse_mnist_tfrec(
                tfrecord, self.name, self.features_shape
            )
        return features, targets
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号