Commit c980eec
- 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 <13048456+shaneahmed@users.noreply.github.com>
1 parent 7fb9926 commit c980eec
File tree
7 files changed
+4572
-4246
lines changed- examples
- full-pipelines
- inference-pipelines
- requirements
- tests/models
- tiatoolbox/models/architecture
7 files changed
+4572
-4246
lines changedLarge diffs are not rendered by default.
0 commit comments