您当前的位置:首页 > IT编程 > Keras[!--writer--]
| C语言 | Java | VB | VC | python | Android | TensorFlow | C++ | oracle | 学术与代码 | cnn卷积神经网络 | gnn | 图像修复 | Keras | 数据集 | Neo4j | 自然语言处理 | 深度学习 | 医学CAD | 医学影像 | 超参数 | pointnet | pytorch |

实现样本数量不均衡批量数据增强 并保存到本地| keras ImageDataGenerator python实例

51自学网 2022-03-01 17:33:41
  Keras
from keras.preprocessing.image import ImageDataGenerator

path = './66' # 类别子文件夹的上一级,如两个分类,子文件夹分别为0,1,他的上一级文件夹为名称为66。如只是增强0文件夹的样本图片,则1文件夹为空即可。
dst_path = './11'
# 图片生成器
datagen = ImageDataGenerator(

    rotation_range=5,
    width_shift_range=0.02,
    height_shift_range=0.02,
    shear_range=0.02,
    horizontal_flip=True,
    vertical_flip=True,
    fill_mode= 'nearest',
    zoom_range=0.2
)

gen = datagen.flow_from_directory(
                           path,
                           target_size=(120, 120),
                           batch_size=142,
                           save_to_dir=dst_path,#生成后的图像保存路径
                           save_prefix='xx',
                           save_format='png')

for i in range(17):
    gen.next()
返回列表
实现批量数据增强 | keras ImageDataGenerator使用
51自学网,即我要自学网,自学EXCEL、自学PS、自学CAD、自学C语言、自学css3实例,是一个通过网络自主学习工作技能的自学平台,网友喜欢的软件自学网站。
京ICP备13026421号-1