이미지 증가

저희는 구글 크롤러를 통해 수집한 약 2000장의 포트홀 이미지를 를 케라스라이브러리를 통해 약 6700장으로 늘렸습니다.

 

이미지를 증가시킬때 필요한 파라미터는 다음과 같습니다.

이미지 증가 코드

import numpy as np
import os
from os import listdir
from os.path import isfile, join
from PIL import Image

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

data_datagen = ImageDataGenerator(rescale=1./255)

data_datagen = ImageDataGenerator(rescale=1./255, 
        rotation_range=90, 
        #shear_range=0.5,
        horizontal_flip=True, 
        vertical_flip=True, 
        fill_mode='nearest')


filename_in_dir = []

for root, dirs, files in os.walk('/home/jong/workingdir/img'):
    for fname in files:
        full_fname = os.path.join(root, fname)
        filename_in_dir.append(full_fname)

for file_image in filename_in_dir:
    print file_image
    img = load_img(file_image)
    x = img_to_array(img)
    x= x.reshape((1,) + x.shape)

    i = 0

    for batch in data_datagen.flow(x,save_to_dir='/home/jong/workingdir', save_prefix='corn_au', save_format='jpg'):
        i += 1
        if i > 6:
            break