python图片数据增强，扩充

最新推荐文章于 2024-09-26 08:30:00 发布

置顶

TimeVshow

最新推荐文章于 2024-09-26 08:30:00 发布

阅读量2.8k

点赞数 5

分类专栏：个人总结文章标签：数据增强

本文链接：https://blog.csdn.net/qq_40184139/article/details/100149901

版权

本文介绍了如何使用Python进行图片数据增强，通过多种技术扩充图像数据集，包括旋转、翻转、裁剪等操作，提升机器学习模型的泛化能力。

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

python图片数据扩充

# coding:utf-8
'''
作者：TimeVShow
time:2019/8/29
效果：找到标签文件中的所有的图片，对其进行水平翻转，垂直翻转等操作，
并保留改变后的图片，同时在标签集中写入图片的信息
'''

import os
from PIL import Image, ImageFilter
import random


def process_train_set(data_dir="./data_sets/cat_12", list_file_name="train_list.txt"):
    file_list_0 = os.path.join(data_dir, list_file_name_0)
    file_list = os.path.join(data_dir, list_file_name)
    with open(file_list, "r+") as flist:
        read_data = flist.read()
        flist.truncate()#清空文件
        for eachline in read_data.split('\n'):
            img_path,label = eachline.split('\t')
            label = label.split("\n")[0]
            img_full_path = os.path.join(data_dir, img_path)
            #如果文件中的图片不存在，直接跳过
            if(os.path.exists(img_full_path) == False):
                continue
            # 进行图像处理和保存，并写入文件
            # 水平翻转

            img_path_temp = horizontalFlip(img_full_path)
            line = img_path_temp.replace(data_dir + "/", "") + "\t" + label + "\n"
            flist.write(line)

            # 垂直翻转

            img_path_temp = verticalFlip(img_full_path)
            line = img_path_temp.replace(data_dir + "/", "") + "\t" + label + "\n"
            flist.write(line)

            # 色彩抖动
            for jitterring_type in range(1, 9)