목표 : 여러개의 사진을 노트북에 업로드해 각 이미지들을 plot 하고 각 이미지에 대한 카테고리 이름을 출력한다.
모델이 n x 299 x 299 x 3 형식의 batch 를 입력받기 때문에, 하나의 이미지라도 이 형태를 띄도록 한다.
특정한 폴더에 이미지를 업로드하면, predict 함수 수행시, best 모델을 로드하고, 이미지를 로드 및 전처리해서 sort 한 순서대로 argmax 로 label 을 구하고 class name 으로 변환한다.
이 class name 과 image 를 같이 조합해서 plot 한다.
'머신러닝 > Korean Food Classifier' 카테고리의 다른 글
[Train] 데이터셋과 모델을 이용해서 훈련 방식을 지정해가면서 훈련하기. (0) | 2022.02.24 |
---|---|
[Model] inception_resnset_v2 직접 구현하기 (0) | 2022.01.31 |
[tf.data] 대규모 데이터셋 파이프라인 구축과정 (0) | 2021.12.23 |