python批量图像上下翻转和旋转90 180 270扩充数据

最新推荐文章于 2024-05-07 14:19:07 发布

深度学习菜鸟

最新推荐文章于 2024-05-07 14:19:07 发布

阅读量3.5k

点赞数 2

分类专栏：图像分类图像增强文章标签： python 深度学习计算机视觉

本文链接：https://blog.csdn.net/qq_36756866/article/details/116450650

版权

图像增强同时被 2 个专栏收录

5 篇文章

订阅专栏

图像分类

2 篇文章

订阅专栏

本文介绍了如何使用Python的PIL库中的Image类和rotate()函数分别进行图像的180度旋转，并展示了通过transpose()方法实现的其他翻转和旋转操作。同时，详细展示了代码示例和txt标签文件的创建过程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

本文介绍的图像的旋转，使用PIL的Image类的接口函数transpose()完成，直接在入参中指定变换方式即可，不仅支持上下、左右翻转；也支持逆时针90、180、270等角度的旋转。

# -*- coding: utf-8 -*-

from PIL import Image
import os

file_dir = 'train_img/png_da_all/'            # 原始图片路径
rotate_180 = 'train_img/rotate_180/'     # 保存路径
    
for img_name in os.listdir(file_dir):
    img_path = file_dir + img_name    #  批量读取图片
    # print(img_path)
    img = Image.open(img_path)
    LEFT_RIGHT = im.transpose(Image.FLIP_LEFT_RIGHT)                    # 左右翻转
    TOP_BOTTOM = im.transpose(Image.FLIP_TOP_BOTTOM)            # 上下翻转
    rotated_90 = img.transpose(Image.ROTATE_90)        # 旋转90度
    rotated_180 = img.transpose(Image.ROTATE_180)    # 旋转180度
    rotated_270 = img.transpose(Image.ROTATE_270)    # 旋转270度
    rotated_180.save(rotate_180 + '180_' + img_name)  # 保存旋转后的图片
    f = open('train_img/180.txt', 'a')                                       
    f.write('180_' + img_name + ' ' + '180' + '\n')        # 写入txt标签文件：imagename + 空格 + 角度
print('finish')

2、图像的旋转使用函数rotate()，在入参中直接指定按逆时针旋转的角度即可，示例如下:

#-*- coding: UTF-8 -*-
from PIL import Image
import os

file_dir = 'train_img/png_da_all/'            # 原始图片路径
rotate_45 = 'train_img/rotate_45/'     # 保存路径

for img_name in os.listdir(file_dir):
    img_path = file_dir + img_name     批量读取图片
    # print(img_path)
    img = Image.open(img_path)
    im_rotate = img.rotate(45)           # 指定逆时针旋转的角度
    im_rotate.save(rotate_45 + '45_' + img_name)  # 保存旋转后的图片