Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

HySpecNet-11k: add new dataset #2410

Merged
merged 7 commits into from
Nov 30, 2024

Conversation

adamjstewart
Copy link
Collaborator

@adamjstewart adamjstewart commented Nov 14, 2024

Adds a data loader for the HySpecNet-11k dataset.

  • Add dataset
  • Add data module
  • Add docs
  • Add tests

enmap

@ando-shah @nilsleh

@github-actions github-actions bot added documentation Improvements or additions to documentation datasets Geospatial or benchmark datasets testing Continuous integration testing datamodules PyTorch Lightning datamodules labels Nov 14, 2024
@adamjstewart adamjstewart added this to the 0.7.0 milestone Nov 19, 2024
@ando-shah
Copy link

There could be a split=None mode which picks up images from all splits. This would be useful for pretraining to maximize sample size

@adamjstewart
Copy link
Collaborator Author

Easiest way to implement that would probably be:

ds = HySpecNet11k(split='train') + HySpecNet11k(split='val') + HySpecNet11k(split='test')

torchgeo/datasets/hyspecnet.py Show resolved Hide resolved
torchgeo/datasets/hyspecnet.py Outdated Show resolved Hide resolved
@adamjstewart adamjstewart merged commit 6baa00d into microsoft:main Nov 30, 2024
19 checks passed
@adamjstewart adamjstewart deleted the datasets/hyspecnet branch November 30, 2024 12:16
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
datamodules PyTorch Lightning datamodules datasets Geospatial or benchmark datasets documentation Improvements or additions to documentation testing Continuous integration testing
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants