* Increase coverage of trainers
* Actually make the tests work
* Updated Cyclone trainer
* Style fix in cyclone tests
* Fixing landcoverai trainer
* Moving mock log to utils
* Fixing the RESISC45 trainer and related
* Skip RESISC45 trainer tests if Windows
* Removing some stupid docstrings from the RESISC45 trainer
* Adding So2Sat trainer tests
* isort
* Adding RESISC45 test data
* Use os.path.join for paths
* Remove unused import
* Add tests for ChesapeakeCVPR trainer
* mypy fixes
* Fix most Chesapeake tests
* Fixed test batching issue in the test dataset
* Get 100% coverage of Chesapeake trainer
* use a FakeTrainer instead of pl.Trainer
* Add naive BYOL trainer tests
* Style fixes
* Add 100% test coverage for BYOL trainer
* Get 100% coverage for LandCover.ai trainer
* Simplify tests
* Add tests for NAIP + Chesapeake trainer
* Fix tests
* Add tests for checkpoint loading
* Reorganize fixtures and specify scope
* Fix various test bugs
* Mypy fixes
* Reduce patch sizes
* Test fewer possible combinations of params
* Prevent warnings in tests
* Restore missing line of coverage in So2Sat trainer
* Silence resampling warning
* Ignore difference in output classes
Co-authored-by: Caleb Robinson <calebrob6@gmail.com>
* 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>
* updated docs
* added VisionClassificationDataset
* refactor PatternNet and RESISC45 to inherit VisionClassificationDataset
* added unit tests and sample data
* refactor PatternNet and RESISC45 to new download/verify checks and fix code coverage
* remove override of __str__
* set default_loader as loader default
* removed loader arg from datasets
* update tests
* format
* remove duplicate code
* updated docstrings
* Added EuroSat dataset
* Cleaning up
* Removing unzipped data
* Added to docs
* EuroSat --> EuroSAT to match paper
* Changing class listing to use bullets
* Update torchgeo/datasets/eurosat.py
Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>
* Addressing review
Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>