欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

图像转成tfrecords很大的问题

程序员文章站 2022-04-30 23:09:50
...

使用默认方式将图像转成tfrecords文件很大

解决方法:构造writer时,加上options参数——

options = tf.python_io.TFRecordOptions(compression_type=tf.python_io.TFRecordCompressionType.ZLIB)
writer = tf.python_io.TFRecordWriter(tfrecords_path, options)

compression_type指定了转换时的压缩方法,ZLIB是一种,还有其它若干种。

读取时,给reader也加上options参数——

options = tf.python_io,TFRecordOptions(compression_type=tf.python_io.TFRecordCompressionType.ZLIB)
reader = tf.TFRecordReader(options=options)
相关标签: TensorFlow 1.X