tensorflow tf.python_io模块源码阅读

最新推荐文章于 2022-09-23 17:58:00 发布

fengwuxuan

最新推荐文章于 2022-09-23 17:58:00 发布

阅读量2.6k

点赞数

CC 4.0 BY-SA版权

分类专栏： tensorflow 文章标签：源码

本文链接：https://blog.youkuaiyun.com/fengwuxuan/article/details/77482404

1.序言
该模块是tensorflow用来处理tfrecords文件的接口，定义在tensorflow/python/lib/io/python_io.py，主要包含了四个部分：
class TFRecordCompressionType：记录的压缩类型。
class TFRecordOptions：用于操作TFRecord文件的选项。
class TFRecordWriter：将记录写入TFRecords文件的类。
tf_record_iterator(…)：从TFRecords文件中读取记录的迭代器

2.源码解析

from __future__ import absolute_import
from __future__ import division
from __future__ import print_function

from tensorflow.python import pywrap_tensorflow
from tensorflow.python.framework import errors
from tensorflow.python.util import compat

#该类定义了tfrecords文件压缩类型：无，ZLIB，GZIP三种
class TFRecordCompressionType(object):
  """The type of compression for the record."""
  NONE = 0
  ZLIB =