build_imagenet_data.py 文件源码

def __init__(self):
    # Create a single Session to run all image coding calls.
    self._sess = tf.Session()

    # Initializes function that converts PNG to JPEG data.
    self._png_data = tf.placeholder(dtype=tf.string)
    image = tf.image.decode_png(self._png_data, channels=3)
    self._png_to_jpeg = tf.image.encode_jpeg(image, format='rgb', quality=100)

    # Initializes function that decodes RGB JPEG data.
    self._decode_jpeg_data = tf.placeholder(dtype=tf.string)
    self._decode_jpeg = tf.image.decode_jpeg(self._decode_jpeg_data, channels=3)

    # Resize
    self._resize = tf.expand_dims(self._decode_jpeg, 0)
    self._resize = tf.image.resize_bilinear(self._resize, [FLAGS.new_height, FLAGS.new_width])
    self._resize = tf.squeeze(self._resize)
    self._resize = tf.cast(self._resize, tf.uint8)

    self._new_jpeg = tf.image.encode_jpeg(self._resize, format='rgb', quality=FLAGS.jpeg_q,
      progressive=False, optimize_size=True, chroma_downsampling=True)