Automatically detect last conv in Grad CAM

sicara / tf-explain

Interpretability Methods for tf.keras models with Tensorflow 2.x

https://tf-explain.readthedocs.io

MIT License

1.02k stars 110 forks source link

Automatically detect last conv in Grad CAM #97

Closed RaphaelMeudec closed 4 years ago

RaphaelMeudec commented 5 years ago

Atm, grad CAM takes a layer_name which is suppose to be the last conv layer of the network. It confuses user, so detecting the last conv automatically would be nice