데브허브 | DEVHUB | [딥러닝 프로젝트] 4강. 사전 훈련된 CNN 모델 | ②강아지와 고양이 사진 분류하기[딥러닝 프로젝트] 4강. 사전 훈련된 CNN 모델 | ②강아지와 고양이 사진 분류하기
- Keras는
keras.applications를 통해 VGG16과 같은 사전 훈련된 CNN 모델을 제공하여 이미지 분류에 쉽게 활용할 수 있습니다. 🧠
- PIL 라이브러리로 이미지를 로드한 후, NumPy 배열로 변환하여 Keras 모델의 입력 형식에 맞춰야 합니다. 🖼️
- 사전 훈련된 CNN 모델은 특정 전처리 방식(예: VGG16의 평균값 빼기)을 요구하며, Keras의
preprocess_input 함수로 이를 편리하게 적용할 수 있습니다. ⚙️
- 모델 예측 시 단일 이미지라도 배치 차원(batch dimension)을 추가해야 하며,
model.predict() 메서드를 사용합니다. 🔢
decode_predictions 함수를 활용하면 모델의 숫자 예측 결과를 사람이 이해하기 쉬운 클래스 이름과 확률로 변환하여 확인할 수 있습니다. 🏷️
- VGG16 모델은 ImageNet 데이터셋으로 훈련되어 1000개의 클래스를 분류하며, 강아지나 고양이 같은 다양한 이미지에 대한 높은 정확도의 예측을 보여줍니다. 🎯