数据扩充

数据扩充

对小图片扩充效果比较好
大图片把不存在的像素用模糊的像素补充

import os 
import shutil as sh 

from keras.preprocessing.image import ImageDataGenerator, array_to_img, img_to_array, load_img

datagen = ImageDataGenerator(
        rotation_range=40,
        horizontal_flip=True,
        vertical_flip=True,
        fill_mode='nearest')



img = load_img("I:/1裂缝检测/CrackForest-dataset/trian/positive2/1320000280.jpg")
x = img_to_array(img)  # this is a Numpy array with shape (3, 150, 150)
x = x.reshape((1,) + x.shape)  # this is a Numpy array with shape (1, 3, 150, 150)

# the .flow() command below generates batches of randomly transformed images
# and saves the results to the `preview/` directory
i = 0
for batch in datagen.flow(x, batch_size=1,
                          save_to_dir='./preview', save_prefix='cat', save_format='jpeg'):
    i += 1
    if i > 10:
        break  # otherwise the generator would loop indefinitely