/GradCAM-CustomData

Primary LanguageJupyter NotebookMIT LicenseMIT

GradCAM-CustomData

  1. custom data를 transfer learning하여 학습하고(ResNet50), 모델 저장

  2. 저장된 모델 load, GradCAM 적용하여 히트맵 출력