머신러닝/Korean Food Classifier

[Predict] 최대 정확도 모델을 통해 사진을 입력받아 예측하기

hwijin97 2022. 2. 24. 15:12
목표 : 여러개의 사진을 노트북에 업로드해 각 이미지들을 plot 하고 각 이미지에 대한 카테고리 이름을 출력한다.

 

모델이 n x 299 x 299 x 3 형식의 batch 를 입력받기 때문에, 하나의 이미지라도 이 형태를 띄도록 한다.

특정한 폴더에 이미지를 업로드하면, predict 함수 수행시, best 모델을 로드하고, 이미지를 로드 및 전처리해서 sort 한 순서대로 argmax 로 label 을 구하고 class name 으로 변환한다.

이 class name 과 image 를 같이 조합해서 plot 한다.