더북(TheBook)

데이터 증가를 위해 이미지를 상하로 이동시켜 봅니다.

코드 5-24 height_shift_range를 이용한 이미지 증가

img_data = expand_dims(data, 0)
data_gen = ImageDataGenerator(height_shift_range=0.5) ------ 이미지 데이터를 상하로 움직일 수 있는 범위 설정
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-37 height_shift_range를 적용한 이미지

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