Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Attention visualization #27

Merged
merged 8 commits into from
Feb 22, 2024
Merged

Attention visualization #27

merged 8 commits into from
Feb 22, 2024

Conversation

pgmikhael
Copy link
Collaborator

  • Add capacity to return attention scores from main model
  • Create function to overlay attention on unprocessed CT scans and save as GIF
  • Amend Serie object loader to read images without processing/normalizing them
  • Update README to reflect changes

@pgmikhael pgmikhael requested a review from jsilter February 7, 2024 16:44
@pgmikhael pgmikhael self-assigned this Feb 7, 2024
@pgmikhael pgmikhael added the enhancement New feature or request label Feb 7, 2024
Copy link
Collaborator

@jsilter jsilter left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Generally looks good to me. I made a few changes in the most recent commit, if those look good to @pgmikhael I'm happy to merge.

* Minor Bugfix visualize_attentions
@jsilter jsilter force-pushed the attention-visualization branch from 37666c1 to da43ce8 Compare February 22, 2024 14:06
@jsilter jsilter merged commit 6453d08 into main Feb 22, 2024
6 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants