AI
[keras_IDG]flow_from_dataframe
교 향
2022. 10. 27. 16:27
Image Data Generator에서 flow_from_directory는 다들 많이 사용하고 본인도 주로 폴더링해서 라벨 구분했었는데
한 데이터로 여러 모델을 만들어야 하고 컬럼 정보가 엑셀로 주어지다 보니 이미지-csv컬럼 속 라벨 간 매칭이 필요했다.
사용법은 flow from directory 랑 비슷한데 옵션을 column요소들을 지정해주면됨
tf.keras.preprocessing.image.ImageDataGenerator | TensorFlow v2.10.0
Generate batches of tensor image data with real-time data augmentation.
www.tensorflow.org
공홈가서 옵션 인자 파악하고 밑의 링크에서 자세한 설명 읽어보면 간단히 사용가능!
Tutorial on Keras ImageDataGenerator with flow_from_dataframe
Keras flow_from_dataframe example article.
vijayabhaskar96.medium.com