더북(TheBook)

데이터를 증가시키기 위해 이미지를 좌우로 이동시켜 봅니다.

코드 5-23 width_shift_range를 이용한 이미지 증가

img_data = expand_dims(data, 0) ------ 차원을 추가하여 데이터 확장
data_gen = ImageDataGenerator(width_shift_range=[-200,200]) ------ 이미지 데이터를 좌우로 움직일 수 있는 범위 설정
data_iter = data_gen.flow(img_data, batch_size=1) ------  iteration 적용
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-36 width_shift_range를 적용한 이미지

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