Skip to content

Conversation

jsilter
Copy link
Collaborator

@jsilter jsilter commented Sep 9, 2024

  • Add example python scripts, both local and remote versions. In both cases the attention is visualized.
  • Bugfix attention visualization
  • Add imageio to list of requirements

Need to clip at 255, not 256. In previous versions anything at 256 would roll over to 0. Affects visualization only, not raw attention calculation.
Make it easier for other tools (eg ark) to calculate attention. Also clip small attention values by default.
…bumentations).

This means we no longer need albumentations or scikit-learn library just for prediction. They would still be used during training (for augmentation) and evaluation.
This is simpler, and consistent between Ark (remote) and local.
@jsilter jsilter self-assigned this Sep 9, 2024
@jsilter jsilter merged commit 479de0a into main Sep 9, 2024
6 checks passed
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.

1 participant