파이썬/텐서플로우

이미지 파일을 학습데이터로 바꿔주는 ImageDataGenerator

공부짱짱열심히하기 2022. 12. 30. 13:15

CNN모델링이 끝난후 사진을 머신러닝에 넣어 분류를 하려고 할때

사진파일은(PNG,JPG등) 넘파이 어레이가 아니기 때문에 그대로 사용이 불가능 하다.

이때 이미지파일을 학습데이터로 변환시켜 주는 작업이 필요하다.

 

이미지 파일을 넘파이 어레이로 바꿔주는 ImageDataGenerator 라이브러리를 사용

 

from keras.preprocessing.image import ImageDataGenerator
train_datagen = ImageDataGenerator(rescale= 1/255.0 )
트레인용
validation_datagen = ImageDataGenerator(rescale= 1/255.0)
밸리데이션 용