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
'AI' 카테고리의 다른 글
| 표준화, 정규화 딱 이해 (0) | 2022.11.18 |
|---|---|
| [accuracy_score] 사이킷런과 케라스의 차이 (0) | 2022.10.31 |
| Early stopping 에서 epoch를 인자값으로 받아올 때 (0) | 2022.10.24 |
| 미드저니 (0) | 2022.09.16 |
| DenseNet (0) | 2022.07.28 |