* Improve sampler performance for pixel-aligned files
* Skip merge if only a single file
* Undo changes to hyperparams
* Fix shape, read all bands
* Remove manual single-file reading
* Always keep workers alive
* Various changes in a desperate attempt to improve performance
* Increase epoch size
* Add missing import, fix model name
* Fix tests
* Persistent workers not used unless entire dataset is consumed
* add RESISC45 trainer
* update working locally
* Adding ability to choose the random split sizes via config
* If you don't have a val or test split, then return the train split so the Trainer doesn't break by default. If you actually want to train without val/test though, then you should set the appropriate Trainer args.
* RESISC experiments
* Reverting accidental changes
* mypy fix
* add dataset_split unit tests
* Document dataset_split
Co-authored-by: Caleb Robinson <calebrob6@gmail.com>
* Added RCF model
* Rename to RCF
* Update tests/models/test_rcf.py
Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>
Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>
* Added support for training on different splits to Chesapeake trainer
* Fixed test script
* Specifying test splits in the test script
* Removing useless prepare_data function from SEN12MS trainer
* Updating LandCoverAI trainer to reflect Chesapeake trainer changes
* Remove verbose in Chesapeake trainer
* Remove adamw in conf files
* Black
* Added experiment run script
* Add augmentation to the landcoverai trainer
* Add augmentation to the landcoverai trainer
* Adding So2Sat trainer
* Fixing style
* Adding So2Sat trainer
* Adding So2Sat trainer
* Adding to so2sat
* Fixing tests
* Add kornia dependency
* Test coverage for GridGeoSampler length
* Added tests for so2sat
* Cleaning up
* Isort
* Adding capability to reindex So2Sat
* Renaming gpu to device
* Setting up so2sat trainer for paper experiments
* So2Sat training sweep
* Idontsort
* Call trainer.tune() step if auto_lr_find is set
* So2Sat trainer work
* Adding SSL method (#162)
* Added support for training on different splits to Chesapeake trainer
* Fixed test script
* Specifying test splits in the test script
* Removing useless prepare_data function from SEN12MS trainer
* Updating LandCoverAI trainer to reflect Chesapeake trainer changes
* Remove verbose in Chesapeake trainer
* Remove adamw in conf files
* Black
* Added experiment run script
* Add augmentation to the landcoverai trainer
* Add augmentation to the landcoverai trainer
* Adding So2Sat trainer
* Fixing style
* Adding So2Sat trainer
* starting SSL trainers
* starting SSL trainers
* Adding BYOL as SSL method, it is fully functional and works for any encoder, next steps includes adding other SSL aproaches like Seasonal Contrast and SSL evaluation at large
* Adding BYOL as SSL method, it is fully functional and works for any encoder, next steps includes adding other SSL aproaches like Seasonal Contrast and SSL evaluation at large
* Making pydocstyle happy
Co-authored-by: Anthony O <anthony.ortiz@microsoft.com>
* Working on experiments
* Updates to So2Sat
* Experiments
* Experiments
* Review changes
* Review changes
* Grabbing the kornia mypy workaround
* updating BYOL to support resnet 50 and updating some docstring
* updating BYOL to support resnet 50 and updating some docstring
* addressing flake8 concerns
* trying to address mypy issues
* mypy
* mypy
* mypy
* adding byol documentation corrections
* Updating byol default and moving experiment scripts to experiments subdir
* Setting up experiment scripts
* Working
* Moving landcoverai seed experiment script
* Fixing formatting in the experiment scripts
* Update torchgeo/trainers/so2sat.py
Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>
* Update torchgeo/trainers/so2sat.py
Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>
* Update torchgeo/trainers/byol.py
Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>
* Update torchgeo/trainers/byol.py
Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>
* Update torchgeo/trainers/byol.py
Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>
* Update torchgeo/trainers/byol.py
Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>
* Update torchgeo/trainers/byol.py
Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>
* Update torchgeo/trainers/byol.py
Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>
* Update torchgeo/trainers/byol.py
Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>
* Update torchgeo/trainers/byol.py
Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>
* Black formatting
* Reorganizing and documenting BYOL code
* Formatting
* BYOL trainer is not Chesapeake specific
* Updating docstrings
* changing to input_channels uniformly
* Fixing BYOL
* Formatting fixes
* Update experiments/run_landcoverai_experiments.py
Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>
* Update experiments/run_landcoverai_experiments.py
Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>
* Update experiments/run_landcoverai_experiments.py
Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>
* Update torchgeo/trainers/byol.py
Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>
* Docstring changes
* Docstring fixes
* Black formatting
* Forward slashes have been judiciously eliminated and I threw in some comments for free
* flake8
* flake8
Co-authored-by: Anthony O <anthony.ortiz@microsoft.com>
Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>
* 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>