더북(TheBook)

데이터를 증가시키기 위해 이미지를 90도 회전시켜 봅니다.

코드 5-26 rotation_range를 이용한 이미지 증가

img_data = expand_dims(data, 0)
data_gen = ImageDataGenerator(rotation_range=90) ------ 이미지 데이터를 90도 회전
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()
i    mage = batch[0].astype('uint16')
    plt.imshow(image)
plt.show()

다음 그림은 이미지가 90도 회전된 결과를 보여 줍니다.

 

 

▲ 그림 5-39 rotation_range를 적용한 이미지

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