더북(TheBook)

데이터 증가를 위해 이미지를 뒤집어(flip) 봅니다.

코드 5-25 flip을 이용한 이미지 증가

img_data = expand_dims(data, 0)
data_gen = ImageDataGenerator(horizontal_flip=True, vertical_flip=True) ------ 이미지 데이터를 뒤집기
data_iter = data_gen.flow(img_data, batch_size=1)
fig = plt.figure(figsize=(30,30))
for i in range(9):
    plt.subplot(3, 3, i+1)
    batch = data_iter.next()
    image = batch[0].astype('uint16')
    plt.imshow(image)
plt.show()

다음 그림은 이미지가 뒤집힌 결과를 보여 줍니다.

 

 

▲ 그림 5-38 flip을 적용한 이미지

신간 소식 구독하기
뉴스레터에 가입하시고 이메일로 신간 소식을 받아 보세요.