Skip to content

Class activation maps for CNN in PyTorch

Compare
Choose a tag to compare
@frgfm frgfm released this 24 Mar 01:31
7be0b4f

This release adds implementations of CAM, GradCAM and GradCAM++.

Note: torchcam 0.1.0 requires PyTorch 1.1 or newer.

brought to you by @frgfm

Highlights

GradCAM

Implementation of gradient-based CAM extractor
New

  • Add a CAM implementation (#2)
  • Add Grad-CAM and Grad-CAM++ implementations (#1, #2).

Test

Verifications of the package well-being before release
New

  • Add test for torchcam.cams (#1, #2)
  • Add test for torschscan.utils (#1)

Documentation

Online resources for potential users
New

  • Add sphinx automatic documentation build for existing features (#1, #2)
  • Add contribution guidelines (#1)
  • Add installation, usage, and benchmark in readme (#1, #2)

Others

Other tools and implementations

  • Add ̀overlay_mask` to easily overlay mask on images (#1).