Repolode
jacobgil

jacobgil/pytorch-grad-cam

컴퓨터 비전 모델의 예측에 영향을 준 이미지 영역을 시각화하는 CAM 기법을 구현한 PyTorch 라이브러리입니다.

스타
13,000
포크
1,715
열린 이슈
166
마지막 커밋
2개월 전

어떤 프로젝트인가요?

pytorch-grad-cam은 PyTorch 기반 컴퓨터 비전 모델을 설명하기 위한 다양한 기법을 제공하는 Python 패키지입니다. 모델 출력에 가장 크게 기여한 픽셀을 보여 주는 히트맵을 생성해 개발 중이나 프로덕션 환경에서 예측 결과를 진단하는 데 도움을 줍니다.

Grad-CAM, HiResCAM, Grad-CAM++, Score-CAM, Ablation-CAM, Eigen-CAM, LayerCAM 등 다양한 CAM 변형을 제공하며, 히트맵의 노이즈를 줄이는 스무딩 옵션과 설명의 충실도를 평가하는 ROAD 같은 지표도 포함합니다. CNN과 Vision Transformer를 지원하며, 분류 외에도 객체 탐지, 시맨틱 세그멘테이션, 임베딩 유사도, CLIP에 적용할 수 있습니다. 배치 입력을 지원하고, reshape transform과 사용자 지정 모델 타깃을 통해 새로운 아키텍처에 맞게 기법을 조정할 수 있습니다.

grad-cam이라는 이름으로 pip를 통해 설치할 수 있습니다. 문서 사이트에는 HuggingFace 분류 모델, Faster R-CNN, YOLO, 세그멘테이션 모델을 다루는 노트북 튜토리얼이 포함되어 있습니다.

핵심 기능

  • 01Grad-CAM, Grad-CAM++, Score-CAM, Ablation-CAM, Eigen-CAM 등 다양한 CAM 기법을 구현합니다.
  • 02ViT와 Swin 같은 CNN 및 Vision Transformer에서 작동합니다.
  • 03객체 탐지, 시맨틱 세그멘테이션, 임베딩 유사도, CLIP을 지원합니다.
  • 04히트맵의 노이즈를 줄이는 스무딩 옵션(aug_smooth 및 eigen_smooth)을 제공합니다.
  • 05설명 품질을 측정하는 ROAD 등의 평가 지표를 포함합니다.
  • 06배치 이미지 처리를 지원하며 CPU, CUDA, MPS 장치에서 실행할 수 있습니다.

활용 사례

  • 이미지 분류기가 특정 예측을 내린 이유를 디버깅할 때.
  • 탐지 또는 세그멘테이션 모델이 올바른 객체에 집중하는지 확인할 때.
  • XAI 연구에서 설명 기법과 지표를 비교할 때.
  • 사용자 지정 데이터셋에서 Vision Transformer의 어텐션과 유사한 동작을 살펴볼 때.

빠른 시작

  1. 1

    PyPI에서 패키지를 설치합니다

    pip install grad-cam
  2. 2

    CAM 메서드와 모델을 가져온 다음 CAM 객체를 만들고 입력 텐서에 실행합니다(사용 예시는 README 참조).

사용 전 알아둘 점

  • PyTorch와 직접 준비한 모델이 필요합니다. 아키텍처마다 올바른 타깃 레이어를 선택해야 합니다.

토픽

자주 묻는 질문

pytorch-grad-cam은(는) 무엇인가요?
컴퓨터 비전 모델의 예측에 영향을 준 이미지 영역을 시각화하는 CAM 기법을 구현한 PyTorch 라이브러리입니다. pytorch-grad-cam은 PyTorch 기반 컴퓨터 비전 모델을 설명하기 위한 다양한 기법을 제공하는 Python 패키지입니다. 모델 출력에 가장 크게 기여한 픽셀을 보여 주는 히트맵을 생성해 개발 중이나 프로덕션 환경에서 예측 결과를 진단하는 데 도움을 줍니다.
pytorch-grad-cam은(는) 무료로 사용할 수 있나요? 라이선스는 무엇인가요?
pytorch-grad-cam은(는) MIT 라이선스의 오픈소스입니다. 상업용 제품에 사용하기 전에 라이선스 조건을 확인하세요.
pytorch-grad-cam은(는) 아직 유지보수되고 있나요?
6개월 이내 업데이트. pytorch-grad-cam의 마지막 커밋은 2026년 8월 13일입니다.
pytorch-grad-cam은(는) 어디에 사용할 수 있나요?
이미지 분류기가 특정 예측을 내린 이유를 디버깅할 때. 탐지 또는 세그멘테이션 모델이 올바른 객체에 집중하는지 확인할 때. XAI 연구에서 설명 기법과 지표를 비교할 때. 사용자 지정 데이터셋에서 Vision Transformer의 어텐션과 유사한 동작을 살펴볼 때.
pytorch-grad-cam는 어떻게 설치하나요?
PyPI에서 패키지를 설치합니다: pip install grad-cam CAM 메서드와 모델을 가져온 다음 CAM 객체를 만들고 입력 텐서에 실행합니다(사용 예시는 README 참조).
pytorch-grad-cam은(는) 어디서 받을 수 있나요?
소스 코드와 설치 방법은 GitHub에 있습니다: https://github.com/jacobgil/pytorch-grad-cam

요약은 프로젝트 README를 바탕으로 AI가 작성했습니다. GitHub 통계는 매일 갱신됩니다.

관련 프로젝트