tensorflow2水平和竖直翻转图片并保存

最新推荐文章于 2023-02-09 17:15:30 发布

吴天德少侠

最新推荐文章于 2023-02-09 17:15:30 发布

阅读量429

点赞数

本文链接：https://blog.csdn.net/sdhdsf132452/article/details/113026737

版权

python 同时被 2 个专栏收录

89 篇文章 3 订阅

订阅专栏

tensorflow2

48 篇文章 2 订阅

订阅专栏

该代码段展示了如何利用TensorFlow库对图像进行左右和上下翻转。首先读取图像文件，然后进行随机翻转操作，接着将翻转后的图像转换回JPEG格式并保存。整个过程遍历指定文件夹内的所有图片，保存翻转后的两个版本。

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

def flip_function(img_path,save_path):
    img = tf.io.read_file(img_path)
    img_original = tf.image.decode_jpeg(img, channels=3)

    img1 = tf.image.random_flip_left_right(img_original)
    img2 = tf.image.random_flip_up_down(img_original)

    img1 = tf.cast(img1, dtype=tf.uint8)
    img2 = tf.cast(img2, dtype=tf.uint8)

    img1 = tf.image.encode_jpeg(img1)
    img2 = tf.image.encode_jpeg(img2)

    img1_save_path = save_path + '_flip1' + '.jpg'
    with tf.io.gfile.GFile(img1_save_path, 'wb') as file:
        file.write(img1.numpy())

    img2_save_path = save_path + '_flip2' + '.jpg'
    with tf.io.gfile.GFile(img2_save_path, 'wb') as file:
        file.write(img2.numpy())


imgs_path = 'C:/Users/a/Downloads/test_images/' # 文件所在文件夹
save_path = 'C:/Users/a/Downloads/test_save/' # 保存文件文件夹

filenames = os.listdir(imgs_path)
for filename in filenames:
    # 要处理的图片路径
    img_path = imgs_path+filename
    # 要保存的图片
    save_path = save_path+filename[:-4]
    flip_function(img_path,save_path)

原始图片：