Skip to content

Conversation

@bwedin
Copy link
Contributor

@bwedin bwedin commented Dec 2, 2022

One-line change: adds
output_layer = output_layer[:,target_class_idx]
to the else statement in def call_model_function. This means that previously Grad-CAM was calculating wrt the entire output layer, this should fix it to only be wrt a single class.

@bwedin bwedin changed the title Update grad-cam to calculate wrt single class Update Examples_core.ipynb grad-cam to calculate wrt single class Dec 2, 2022
@gkapish gkapish self-requested a review December 2, 2022 18:16
@gkapish gkapish self-requested a review December 2, 2022 19:17
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants