Граф коммитов

5 Коммитов

Автор SHA1 Сообщение Дата
Yi-Chia Chang a757cf14fb
SatlasPretrain: ResNet50/152 and Swin_V2_T Weights (#2038)
* add SENTINEL2_MS_MI_SATLAS

* add SENTINEL2_MS_SI_SATLAS

* fix style

* ruff

* Add all SatlasPretrain models

* Get local tests passing

* Get local tests passing

* Fix remote tests

* Mock input channels in testing too

* Fix license

* Fix band order

---------

Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>
2024-08-27 15:58:49 +02:00
Isaac Corley 8d137526f7
Scale-MAE model (#2057)
* add scale-mae model and pretrained weights

* fix docs, mypy, and code coverage

* add scalemae to torch hub

* make res optional instead of required

* Remove ViT from the name

* Alphabetical order

* Doc bugs

---------

Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>
2024-08-19 18:50:36 +02:00
Adam J. Stewart dbfe7fa864
torch.hub: add support for DOFA and Swin models (#2052)
* torch.hub: add support for DOFA and Swin models

* Fix tests

* Add *args support to DOFA to match other models
2024-05-12 09:28:28 +02:00
Adam J. Stewart b09122f3e4
Ruff: prefer single quotes over double quotes (#2001)
* Ruff: prefer single quotes over double quotes

* Apply ruff format
2024-05-03 19:30:14 +02:00
Nils Lehmann 60eb61b5fa
Add Multi-Weight Support API (#917)
* load pretrained weights

* change name millionaid

* restructure and additional weights

* rename sentinel1 weights

* add vit small weights

* forgot to add vit.py

* struggling with test

* wrong name failing test

* feedback on tests

* increase test coverage

* fix failing test

* fix failing test

* fix failing test and add vit tests

* fix failing vit test

* torchgeo.models.utils

* forgot utils file

* typo num channels

* nitpick docs, version torchvision

* another try min dependencies

* add documentation table

* expand pytests to test pretrained weights on tasks

* reverse changes to byol task

* add tests to init pretrained weights from config

* forgot to add the conf files

* change path

* increase test coverage

* vit tests all pass locally including slow

* now remote

* fix tests another one

* add a draft tutorial

* run black on tutorial notebook

* Tutorial typo fixes

* Lower min torch/vision versions

* Fix bad rebase

* Remove dead code

* Flake8 fixes

* Consistent in_chans

* Black fixes

* bison > yacs

* Remove one more reference

* Download modified weights from hugging face

* Add entrypoints

* Add torch.hub support

* progress arg is required

* Fix model loading for resnet18

* Add transforms, update tests

* VIT -> ViT

* add seco weights

* Fix type hints

* Link to timm docs

* Fix pydocstyle

* Try to fix timm docs link

* Fix tests

* Nuke ignores

* Ignore timm links

* Add model API methods

* Add to __init__ and document

* Test model API functions

* fix tests

* Use correct documentation link for intersphinx

* Typos

* Fix Windows tests

* meth -> func

* Explicit function scope

* weight-specific filename

* Support enums in classification trainer

* Update other trainers too

* Fix regression tests

* Fix classification tests

* Fix byol tests

* Fix types

* progress_bar is required arg

* Test weight enums

* Fix pickling

* Fix regression tests

* Improve coverage of classification tests

* Improve coverage of BYOL tests

* Update resnet table

* Update ViT table

* Update get_state_dict usage

* Remove unused YAML files

* Update table widths

* Documentation improvements

* Tweak tables

* Try to fix Windows tests

* Revert "Try to fix Windows tests"

This reverts commit 1325b13ff7.

* Monkeypatch everything

* Revert "Monkeypatch everything"

This reverts commit e3e8d7d042.

* Revert "Revert "Monkeypatch everything""

This reverts commit 9b27bd705b.

* Patch things not at the source

* Fix missing import

Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>
2023-01-22 14:25:49 -08:00