build_imagenet_data.py 文件源码

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

项目:keras_experiments 作者: avolkov1 项目源码 文件源码
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)
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号