Ultralytics YOLOv5는 객체 탐지, 인스턴스 분할, 이미지 분류를 지원하며 학습과 내보내기를 제공하는 PyTorch 기반 컴퓨터 비전 툴킷입니다.
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
PyPI에서 패키지를 설치합니다
pip install grad-cam - 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 통계는 매일 갱신됩니다.
관련 프로젝트
코드북 조회 트랜스포머로 사진과 동영상의 흐리거나 품질이 낮고 손상된 얼굴을 복원하는 Python 도구
PyTorch 기반 딥러닝 모델로 PDF와 이미지의 텍스트를 탐지하고 인식하는 오픈소스 Python OCR 라이브러리입니다.
고해상도 이진 이미지 분할과 배경 제거에 널리 쓰이는 딥러닝 모델 BiRefNet의 공식 PyTorch 구현입니다.
CVPR 2025 LiveCC 논문의 실시간 영상 해설을 생성하는 스트리밍 음성 전사 기반 학습 비디오 LLM입니다.
대규모 모델의 분산 학습과 추론을 더 쉽고 효율적으로 지원하는 딥러닝 최적화 라이브러리입니다.