tf图像相关处理

"""
图像的预处理过程
"""
import os
import matplotlib.pyplot as plt
import tensorflow as tf


# 读取数据
def get_decode_image(file_name, dtype=tf.uint8, has_eval=False):
    # 读取原始数据
    image_raw_data = tf.gfile.FastGFile(file_name, "rb").read()

    # 解码图片数据得到三维矩阵
    # image_data = tf.image.decode_jpeg(image_raw_data)
    # image_data = tf.image.decode_png(image_raw_data)
    image_data = tf.image.decode_image(image_raw_data)

    # 转换数据类型
    image_data = tf.image.convert_image_dtype(image_data, dtype=dtype)

    if has_eval:
        with tf.Session():
            image_data = image_data.eval()

    return image_data


# 由于图片存在压缩，将一张图像还原成三维矩阵，需要进行解码。
def decode_image_and_save(file_name="data_process.jpg"):
    name, ext = os.path.splitext(file_name)
    # 读取数据
    image_data = get_decode_image(file_name)

    # 编码图片
    encoded_image = tf.image.encode_jpeg(image_data) if ext.lower() == "png" else tf.image.encode_png(image_data)
    ext = "jpg" if ext.lower() == "png" else "png"

    with tf.Session():
        # 保存图片
        with tf.gfile.GFile(name="{}.{}".format(name, ext), mode="wb") as f:
            f.write(encoded_image.eval())

        plt.imshow(image_data.eval())
        plt.show()
    pass


# 图片大小调整:通过算法使新的图像尽量保存原始图像的所有信息。
def resize_image(file_name="data_process.jpg"):
    # 读取数据
    image_data = get_decode_image(file_name, dtype=tf.float32, has_eval=True)

    # 1.调整大小：插值
    resized_images = tf.image.resize_images(image_data, [300, 300], method=tf.image.ResizeMethod.BILINEAR)
    # 2.调整大小：crop or pad
    # resized_images = tf.image.resize_image_with_crop_or_pad(image_data, target_height=500, target_width=500)
    # 3.调整大小：比例
    # resized_images = tf.image.central_crop(image_data, 0.5)
    # 4.调整大小：裁剪
    # resized_images = tf.image.crop_to_bounding_box(image_data, 100, 100, target_height=400, target_width=500)
    # 5.调整大小：填充
    # resized_images = tf.image.pad_to_bounding_box(image_data, 400, 100, target_height=1000, target_width=1200)
    # 6.调整大小：裁剪并resize
    # image_data = tf.expand_dims(image_data, axis=0)
    # resized_images = tf.image.crop_and_resize(image_data, [[0.8, 0.2, 0.5, 0.5]], box_ind=[0], crop_size=[200, 200])
    # resized_images = tf.squeeze(resized_images, axis=0)

    print(resized_images.get_shape())

    # 显示图片
    with tf.Session():
        plt.imshow(resized_images.eval())
        plt.show()
    pass


# 图像翻转
def flip_image(file_name="data_process.jpg"):
    image_data = get_decode_image(file_name, dtype=tf.float32)

    # 翻转
    # flip_image_data = tf.image.flip_up_down(image_data)
    # flip_image_data = tf.image.flip_left_right(image_data)
    # 随机翻转
    # flip_image_data = tf.image.random_flip_left_right(image_data)
    # flip_image_data = tf.image.random_flip_up_down(image_data)
    # 转置
    flip_image_data = tf.image.transpose_image(image_data)

    print(flip_image_data.get_shape())

    with tf.Session():
        plt.imshow(flip_image_data.eval())
        plt.show()
    pass


# 图像色彩调整:亮度、对比度、色相、饱和度
def adjust_color(file_name="data_process.jpg"):
    image_data = get_decode_image(file_name, dtype=tf.uint8, has_eval=True)

    # 亮度
    adjust_image_data = tf.image.adjust_brightness(image_data, 0.5)
    # adjust_image_data = tf.image.random_brightness(image_data, 0.5)

    # 对比度:(x - mean) * contrast_factor + mean
    # adjust_image_data = tf.image.adjust_contrast(image_data, 5)
    # adjust_image_data = tf.image.adjust_contrast(image_data, -5)
    # adjust_image_data = tf.image.random_contrast(image_data, 2, 5)

    # 色相
    # adjust_image_data = tf.image.adjust_hue(image_data, 0.5)
    # adjust_image_data = tf.image.adjust_hue(image_data, -0.5)
    # adjust_image_data = tf.image.random_hue(image_data, 0.4)

    # 饱和度
    # adjust_image_data = tf.image.adjust_saturation(image_data, 5)
    # adjust_image_data = tf.image.adjust_saturation(image_data, 0.4)
    # adjust_image_data = tf.image.random_saturation(image_data, 0.4, 10)

    # 标准化:(x - mean) / adjusted_stddev
    # adjust_image_data = tf.image.per_image_standardization(image_data)

    with tf.Session():
        plt.imshow(adjust_image_data.eval())
        plt.show()
    pass


# 标准化
def standard_image(file_name="data_process.jpg"):
    image_data = get_decode_image(file_name, dtype=tf.uint8, has_eval=True)

    # 标准化:(x - mean) / adjusted_stddev
    standard_image_data = tf.image.per_image_standardization(image_data)

    with tf.Session():
        print(standard_image_data.eval())
    pass


# 处理标注框
def draw_bounding_boxes(file_name="data_process.jpg"):
    image_data = get_decode_image(file_name, dtype=tf.float32, has_eval=True)
    image_data = tf.image.resize_images(image_data, [500, 500], method=tf.image.ResizeMethod.BILINEAR)
    image_data = tf.expand_dims(image_data, axis=0)
    result = tf.image.draw_bounding_boxes(image_data, boxes=[[[0.1, 0.1, 0.2, 0.2], [0.3, 0.3, 0.5, 0.5]]])
    result = tf.squeeze(result, axis=0)
    with tf.Session():
        plt.imshow(result.eval())
        plt.show()
    pass

if __name__ == '__main__':
    draw_bounding_boxes()