Skip to content

Commit

Permalink
🆕 Integrate Foundation Models Available VIA timm: UNI, `Prov-Giga…
Browse files Browse the repository at this point in the history
…Path`, `H-optimus-0` (#856)

- Integrates pre-trained foundation models from other labs into tiatoolbox.models.architecture.vanilla.py.

Currently, the `_get_architecture()` function allows the use of models from `torchvision.models`.

A new function `_get_timm_architecture()` has been incorporated to support foundation models which are available from `timm` with weights on HuggingFace Hub. 

- All the models from `timm` that used require users to sign the licence agreement with the authors.
- The users can add new models by modifying `_get_timm_architecture()`

---------

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Shan E Ahmed Raza <[email protected]>
  • Loading branch information
3 people authored Nov 15, 2024
1 parent 7fb9926 commit c980eec
Show file tree
Hide file tree
Showing 7 changed files with 4,572 additions and 4,246 deletions.
86 changes: 57 additions & 29 deletions examples/03-tissue-masking.ipynb

Large diffs are not rendered by default.

Loading

0 comments on commit c980eec

Please sign in to comment.