search

utkuozbulak / pytorch-cnn-visualizations

Pytorch implementation of convolutional neural network visualization techniques
MIT License
7.81k stars 1.49k forks source link

Support for LayerCAM #101

Closed PengtaoJiang closed 3 years ago

PengtaoJiang commented 3 years ago

Hi, @utkuozbulak, Our paper "LayerCAM: Exploring Hierarchical Class Activation Maps for Localization" is accepted by TIP recently, which can visualize the class activation maps from any cnn layer of an off-the-shelf network. Could you add our method to your popular repository for more people to try this method? Our method is a simple modification of Grad-CAM. It should easy to implement. Here is the paper and code. Hope for your reply.

utkuozbulak commented 3 years ago

Hey, feel free to send a PR!

PengtaoJiang commented 3 years ago

Thanks a lot!

PengtaoJiang commented 3 years ago

@utkuozbulak I have sent you a PR! Feel free to check it.

utkuozbulak commented 3 years ago

Yeah, will have a look as soon as there is spare time. Thanks for the PR.