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

Add a num_classes option to test script, update README #3

Merged

Conversation

metazool
Copy link
Contributor

Hello!

This contains some more small tweaks needed to run all the models included in the Google Drive linked to the project.

  • Adds an -n / --num_classes option defaulting for 3
  • Adds the set of 18 class labels for the models that support them

See #1 for setup and for the commandline options used, which are also documented in the README.

I'd definitely offer some small suggestions for improving the reuse potential of this work - that probably begins with finding a FAIR home for the model weights and associated model card, though - see #2 (I can try following this up through the scivision front door!)

@noushineftekhari noushineftekhari merged commit 563df1f into alan-turing-institute:main Nov 15, 2024
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