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

16 Коммитов

Автор SHA1 Сообщение Дата
Nils Lehmann b2b651644c
Add CROMA pretrained model (#2370)
* add croma

* coverage

* single line test

* Fix type hints

* review

* Update croma.py

Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>

* review

* typo

---------

Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>
2024-10-29 09:36:36 +01:00
Adam J. Stewart 7ad9df8f32
Docs: emphasize multi-modal foundation models (#2236)
* Docs: emphasize multi-modal foundation models

* Subdirectory for weights files

* Update col widths
2024-08-27 16:20:18 +02:00
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
XShadow 2849944a02
Add DOFA model (#1903)
* adding DOFA-Net model

* Convert git submodule to single file

* Get rid of __main__ code blocks

* Add WeightsEnum

* Add documentation

* Add CSV file

* Update link to transform implementation

* Add unit tests

* Test model forward function

* Complete test coverage

* Add type hints

* Solve most type issues

* Solve remaining type issues

* Undo sorting

* Simplifed docs

* Fix loading of real weights

* OFAViT -> DOFA

* Remove redundant helper function

* Add units for wavelengths

* wave_list -> wavelengths

* wvs -> wavelengths

* img_feat -> x

* Sorting

* inter_dim is not used

* Rename embedding layer

* Simpler name for position embedding

* wv_planes -> dynamic_embed_dim

* make weight init a hidden method

* Simpler model init

* Use permalink

* Document GEO-Bench performance

* Simpler test

* More columns

* Update correct column numbers

* Modified datasets

* Add large weights

* Update __all__

---------

Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>
2024-03-25 11:44:31 +00:00
Kaleb Crans 19b723fa30
Added license column to dataset csv files (#1732)
* Added license column to dataset csv files

* fixed column formatting in rst files

* fixed typo for apache licenses

* fixed license for landsat model weights

* fixed public domain licenses and added new licenses

* Apply first set of suggestions from code review

Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>

* Kenya Crop Type: fix license

---------

Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>
2023-12-08 07:42:44 -06:00
Isaac Corley 8c24c80bb3
Satlas pretrained Swin Transformer v2 models (#1358)
* add satlas pretrained models

* update swin_b -> swin_v2_b

* update docs

* swin_b -> swin_v2_b again

* add the tests

* tests

* fix isort error

* update weights csv files

* swin_v2_b added in torchvision 0.14 so update min reqs

* fix swin_v2_b weights api

* torchvision 0.14.0 depends on torch 1.13.0

* fix docs

* fix docs

* fix swin docs

* update comment and import from root module

* update torch/torchvision min versions

* Move to satellite-specific tables

* Move to satellite-specific tables

---------

Co-authored-by: Caleb Robinson <calebrob6@gmail.com>
Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>
2023-11-25 20:39:43 +00:00
Nils Lehmann 18139b440b
SSL4EO-L Weights (#1482)
* ssl4eo landsat 18 weights

* add resnet and vit moco weights under new hf repo

* add simclr ssl checkpoint

* add docs

* order models and new csv tablesf for docs

* adjust spacing of table

* lots of numbers and formatting

* missing resnet etm-sr weights

* fix failing tests

* check slow tests

* Fix URL checksum

* fix weight names

* Use 2 digits for all metrics

* hyphen -> en dash

* Use 2 digits for all metrics

* Landsat first

* EOL char before EOF char

---------

Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>
2023-09-30 13:50:43 -05: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
Adam J. Stewart b412f52efc
Docs: shorter dataset/datamodule/model names (#571) 2022-06-15 14:00:46 -07:00
isaac 2c6e7eea60
FCCD Models refactor (#345)
* remove FCEF model, refactor FCSiamConc and FCSiamDiff to inherit from smp.Unet

* style fixes update tests

* mypy and docstring fixes

* more mypy fixes and add tests

* remove test args

* fix tests

* add smp to intersphinx mapping

* update model DOI

* add the right docs this time

* Removing type ignores

* Testing sphinx fix

* Added parts of SMP to nitpicky ignore

* Fixing docs

Co-authored-by: Caleb Robinson <calebrob6@gmail.com>
2022-02-15 22:54:52 +00:00
Adam J. Stewart 0fbe42b465
Add ResNet pre-trained models to the docs (#256)
* Add ResNet pre-trained models to the docs

* Fix missing reference link

* ResNet is not documented, nothing to link to
2021-11-29 07:01:54 -08:00
Caleb Robinson f68cec5a2e
Docs/models (#199)
* Making sphinx build in parallel

* Cleaning up the model docs
2021-10-17 12:14:26 -05:00
Caleb Robinson 56e300929f License removal team 2021-08-31 13:25:05 -05:00
Adam J. Stewart da7e120350 Add license header to all files 2021-08-31 13:25:05 -05:00
Adam J. Stewart 36c3e17154 Add tutorials to documentation 2021-08-25 11:19:33 -05:00