使用tensorflow对图像数据增强和增广

最新推荐文章于 2022-10-20 18:05:20 发布

上进的小菜鸟

最新推荐文章于 2022-10-20 18:05:20 发布

阅读量1.5k

点赞数 2

分类专栏： tensorflow 数据增强文章标签：图像数据增强图像数据扩增翻转旋转 tensorflow

本文链接：https://blog.csdn.net/qq_35153620/article/details/102968688

版权

该博客介绍了如何利用TensorFlow库对图像数据进行增强和扩增，包括图像的垂直翻转、水平翻转以及按90度倍数的旋转操作，旨在提升深度学习模型的训练效果。

摘要由CSDN通过智能技术生成

使用tensorflow

分别对数据集中图像使用翻转(垂直和左右)和旋转(90度的倍数)

import tensorflow as tf
import os

def trans(img, img_data, sav):
    #翻转操作是镜像的，旋转是非镜像的
    #垂直翻转
    flipped_up_down = tf.image.flip_up_down(img_data)
    #水平翻转
    flipped_left_right = tf.image.flip_left_right(img_data)
    #以90度的倍数进行旋转
    rotate_90 = tf.image.rot90(img_data, k=1)
    rotate_180 = tf.image.rot90(img_data, k=2)
    rotate_270 = tf.image.rot90(img_data, k=3)
    #编码
    encoded_image_u = tf.image.encode_jpeg(flipped_up_down)
    encoded_image_l = tf.image.encode_jpeg(flipped_left_right)
    encoded_image_r9 = tf.image.encode_jpeg(rotate_90)
    encoded_image_r18 = tf.image.encode_jpeg(rotate_180)
    encoded_image_r27 = tf.image.encode_jpeg(rotate_270)
    with tf.Session() as sess:
        u, l, r9, r18, r27 = sess.run([encoded_image_u, encoded_image_l, encoded_image_r9, 
                encoded_ima