* Debug failing notebook tests
* Known working nbmake
* Remove buggy tutorial
* Undo version change
* No longer need to rerun tests
* Update tutorials to new syntax
* 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>
* adding QR loss functions for learning on the prior
* chesapake learn on prior trainer with self-contained code for visualization
* adding prior dataset to the chesapeake datasets; doesn't handle downloading or anything like that yet
* updating init files to include chesapeake CVPR prior
* adding FCNModified for learning on the prior
* changing input to samplers to pass dataset instead of dataset.index
* fixing style issues
* Removing FCN_modified
* Fixing super call and mypy in FCN model
* Added learning on the prior extension
* Update tests
* Formatting
* Adding QR loss
* Added losses to docs
* Removing trainer, moving datamodule
* Combining chesapeake and chesapeake prior datamodules
* Formatting
* Test coverage
* Formatting
* Adding losses
* Re-moving the datamodules around
* Make loss function a torch Module
* Version added
* Fixed some stuff that got messed up in the rebase
* Formatting
* How'd this get there?
* Change qr losses to expect probabilities instead of log-probabilities
* Clean up test
* Rename qr loss file
* Renamed test file
Co-authored-by: Caleb Robinson <calebrob6@gmail.com>
* Add custom RasterDataset notebook
* Update docs index.rst
* Update copyright, fix URL typo, and add verbose description
* Add xview3 sample data
* Update notebook
* Show simple example first, complicated example second
* Remove the second half of the notebook, can expand later
* draft indices transform
* import sort
* added AugmentationSequential wrapper for dicts
* updated indices
* fix dim concat bugs
* format
* add kornia dependency
* add augmentationsequenal unit tests
* add augmentationsequential support for boxes and mask dtypes
* add indices tests
* Draft indices tutorial notebook
* move notebook to tutorials folder
* mypy fixes
* fix bug when only image key used in AugmentationSequential
* Created using Colaboratory
* added tutorial to docs
* format
* added kornia master branch dependency'
* refactor notebook to use % cell magic and python to download files
* revert kornia version
* install kornia master branch for mypy checks
* update mypy github action install order
* fix divide by zero error in indices
* Created using Colaboratory
* fix nbsphinx errors
* add TODO to remove kornia in tests action
* format setup.cfg
* minor fixes to indices
* remove unecessary variable
* update mask to cast to original dtype
* removed unused ignore comment
* added gray/rgb/multispectral unit tests
* added tests with boxes
* Created using Colaboratory
* Created using Colaboratory
* fix mypy issues
* updated notebooks in docs
* Updates to tutorials
* Created using Colaboratory
* Created using Colaboratory
Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>
* Moving task specific configuration logic from train.py into respective classes
* Small fixes
* Adding basic FCN model for benchmarking
* Adding simple FCN model
* Removing OrderedDict from model definitions
* Adding torchgeo.models to docs
* Adding model tests
* Making all the formatters happy
* Adding optimizer options to landcoverai
* Fixing conda environment I think
* How do you feel about a Makefile, Adam?
* Formatting
* Adding some documentation to the readme
* Sanity check command in README
* Fixes in the landcoverai datamodule to make multi-GPU training possible
* Closing figures that we send to Tensorboard
* Fix sphinx missing target warning
* Fix pytest coverage
* Fix flake8
* Update torchgeo/models/__init__.py
Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>
Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>
* Initial commit of lightning based model training framework
* Made save directories work correctly
* Add pytorch-lightning dependency and some comments
* More documentation and cosmetic tweaks
* Typo fix
Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>
* Fix some style issues
* Fix pydocstyle
* Add missing sklearn dependency
* Try to get conda environment working
* Add documentation
* Ignore missing target reference
* Make train.py executable
* Ignore logs and output dirs
* Raise exceptions instead of returning
* Move all argparse stuff to set_up_parser
* Add tests for train.py
* Fix Python 3.6 compatibility
* Fix support for older versions of pytorch-lightning
Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>
Adam J. Stewart
Nils Lehmann
estherrolf
Ritwik Gupta
isaac
Caleb Robinson