* Update pretrained_weights.ipynb
Fixed an error in the state dict loading of the turorial and added a comment on the num_classes parameter when creating timm models.
* Update docs/tutorials/pretrained_weights.ipynb
* Update utils.py
* Import Tuple from typing
* Change return of `load_state_dict` from `model` to `Tuple[List[str], List[str]]`, matching the return of the standard PyTorch builtin function.
* Update pretrained_weights.ipynb
Remove example of loading pretrained model without prediction head (`num_classes=0`).
* Update README.md
Adapt new `load_state_dict` function.
* Mimic return type of builtin load_state_dict
* Modern type hints
* Blacken
* Try being explicit
---------
Co-authored-by: Caleb Robinson <calebrob6@gmail.com>
Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>
* paragraph about pretrained models
* paragraph about pretrained models
* suggested changes
* Update README.md
Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>
---------
Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>
* add pretrained weights loading for the segmentation encoder
* Updating config files to use new pretrained arg style
* fix loading weights enum to encoder
* I have no idea what I'm doing with these tests
* tests passing
* update docstring
* add the tests back in dummy
* add tests
---------
Co-authored-by: Caleb Robinson <calebrob6@gmail.com>
* Bump pytorch-lightning[extra] from 1.9.4 to 2.0.0 in /requirements
Bumps [pytorch-lightning[extra]](https://github.com/Lightning-AI/lightning) from 1.9.4 to 2.0.0.
- [Release notes](https://github.com/Lightning-AI/lightning/releases)
- [Commits](https://github.com/Lightning-AI/lightning/compare/1.9.4...2.0.0)
---
updated-dependencies:
- dependency-name: pytorch-lightning[extra]
dependency-type: direct:production
update-type: version-update:semver-major
...
Signed-off-by: dependabot[bot] <support@github.com>
* Remove deprecated features
* pytorch_lightning -> lightning
* bump radiant-mlhub too
* isort fix
* lightning does not yet have public type hints
* More ignores
* blanket ignore
* blanket ignore
* Fix function typo
---------
Signed-off-by: dependabot[bot] <support@github.com>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>
Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>
* File rename: nwpu.py -> vhr10.py
* Update more locations
* Name change in several other places
* Add NWPU only to docstring
Co-authored-by: Caleb Robinson <calebrob6@gmail.com>
* Adding a UnionDataset
* Adding contains method to BoundingBox
* Finishing UnionDataset
* Add __contains__ method
* Overhaul BoundingBox, add set arithmetic
* mypy fixes
* pydocstyle fixes
* Ignore erroneous pydocstyle warnings
* rtree only supports tuples, not BoundingBoxes
* mypy fixes
* Use custom collate function to handle BoundingBoxes
* Add back support for Python 3.6
* Add tests for all new BoundingBox features
* Rename ZipDataset to IntersectionDataset
* Merge indices of IntersectionDataset, auto-convert CRS/res
* Get tests to pass
* Fix more tests
* Test more of RasterDataset/VectorDataset directly
* Increase UnionDataset test coverage
* IntersectionDataset stacks tensors, UnionDataset merges tensors
* Support collating dicts with differing keys, add tests
* Style fixes
* Samplers: compute intersection between index and ROI
* Update README with example usage
* GeoDataset addition is deprecated
* Add note about CRS/res
* More documentation for Intersection/UnionDatasets
* Use collate function in tutorial
* Don't use multiple workers
* Fix typo
* Drop support for adding GeoDatasets
* Remove unused import
* Add comment explaining coverage config settings
* Collation function needed for benchmark script
* Add more explanation to README
* Correct Landsat 8 bands
* Print warning when changing CRS/res
Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>
* 0.1.0 release
* Train deps needed for release testing
* Update development status
* setup.py should not be run directly
* Test more trainers
* Fix local docs build
* Update installation instructions
* Specify test data dir in config
* Fix tutorial docs
* Trainers should default to num_workers=0, download=False
* Correct location for root_dir
* Try different GDAL name
* Try again
* Various fixes to release tests
* Update pip installs in tutorials
* Fix some bugs
* Config file not being picked up
* Get back to 100% test coverage
* Added correct weight string to UCMerced
* yolo fix
* yolo fix pt 2
* yolo fix 2 pt. 1
* Simplify tests a bit
* Make the trainer notebook look stupid
* UCMerced should download by default in the trainers
* Revert
* Fix logo/author, include LICENSE in upload
Co-authored-by: Caleb Robinson <calebrob6@gmail.com>
Adam J. Stewart
Yi-Chia Chang
Oluwabusayo Jacobs
Slawek Bierwiaczonek
Ashwin Nair
Konstantin Klemmer
Nils Lehmann
Robin Cole
Burak
Isaac Corley
dependabot[bot]
Caleb Robinson