Python批量reshape图片

代码如下:

# -*- coding: utf-8 -*-
"""
Created on Thu Aug 23 16:06:35 2018

@author: libo
"""
from PIL import Image
import os


def image_resize(image_path, new_path):           # 统一图片尺寸
    print('============>>修改图片尺寸')
    for img_name in os.listdir(image_path):
        img_path = image_path + "/" + img_name    # 获取该图片全称
        image = Image.open(img_path)              # 打开特定一张图片
        image = image.resize((512, 512))          # 设置需要转换的图片大小
        # process the 1 channel image
        image.save(new_path + '/'+ img_name)
    print("end the processing!")


if __name__ == '__main__':
    print("ready for ::::::::  ")
    ori_path = r"Z:\pycharm_projects\ssd\VOC2007\JPEGImages"                # 输入图片的文件夹路径
    new_path = 'Z:/pycharm_projects/ssd/VOC2007/reshape'                   # resize之后的文件夹路径
    image_resize(ori_path, new_path)

 

数据增强是指在不改变标签的情况下,对已有的数据进行一定的变换,从而生成新的数据,增加数据样本的多样性和数量,提高模型的泛化能力。以下是一个基于Python的数据增强代码示例: ```python import os import cv2 import numpy as np from keras.preprocessing.image import ImageDataGenerator # 定义数据增强的方法 def data_augmentation(): datagen = ImageDataGenerator( rotation_range=10, # 随机旋转的角度范围 width_shift_range=0.1, # 水平方向上平移的范围 height_shift_range=0.1, # 垂直方向上平移的范围 shear_range=0.2, # 剪切变换的程度 zoom_range=0.2, # 随机缩放的范围 horizontal_flip=True, # 水平翻转 fill_mode='nearest' # 填充像素的方式 ) return datagen if __name__ == '__main__': data_path = 'path/to/your/data' save_path = 'path/to/save/data' if not os.path.exists(save_path): os.makedirs(save_path) image_list = os.listdir(data_path) for image_name in image_list: image_path = os.path.join(data_path, image_name) img = cv2.imread(image_path) img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB) # 转换为RGB格式 img = img.reshape((1,) + img.shape) # 扩展为4维数组 datagen = data_augmentation() # 定义数据增强器 save_prefix = image_name.split('.')[0] # 利用数据增强器生成新的数据 for i, new_image in enumerate(datagen.flow(img, batch_size=1, save_to_dir=save_path, save_prefix=save_prefix, save_format='jpg')): if i >= 9: # 生成10张新图片 break ``` 以上代码使用Keras中的`ImageDataGenerator`方法定义了数据增强器,并通过调整不同的参数来实现旋转、平移、缩放、剪切等变换。在主函数中,读取原始图片文件夹中的图片,然后利用`ImageDataGenerator`生成新的图片,并保存到指定的文件夹中。生成的新图片文件名以原始图片文件名为前缀,后接一个数字表示第几张新图片
评论 7
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值