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

Add plot method and new band names for Sen12MS dataset #320

Closed
wants to merge 27 commits into from

Conversation

nilsleh
Copy link
Collaborator

@nilsleh nilsleh commented Dec 21, 2021

Add plot method to Sen12MS dataset, as well as passing band names as Tuple(str) as mentioned in #289.

TODO/Questions:

  • the mask in this dataset has 4 channels that represent MODIS Land Cover as I understand from here, how should this be handled in the plot?
  • show example that plot method yields
  • looking at the lighting module, should both passing a tuple of strings, as well as a key word like "all", "s1", "s2" be supported?
  • Sen12ms Dataset repo mentions that there is also an extension of this dataset with a classification task. Is that something that should be added?

@adamjstewart adamjstewart added the datasets Geospatial or benchmark datasets label Dec 21, 2021
@adamjstewart adamjstewart added this to the 0.2.0 milestone Dec 21, 2021
@adamjstewart adamjstewart mentioned this pull request Dec 21, 2021
19 tasks
adamjstewart and others added 18 commits December 21, 2021 14:05
…lue (microsoft#277)

* Various improvements to API docs

* Remove setting that is already the default
* 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
* Move DataModules to torchgeo.datamodules

* Clean up local imports
* plotting method cowc dataset

* boolean label for detection

* restructure plotting method

* typo

* label title not as variable

* single plot method in super class

Co-authored-by: Caleb Robinson <[email protected]>
…PR data (#202)

* 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 <[email protected]>
…datasets (microsoft#264)

* Adding plot to ADVANCE dataset

* Adding plot to BeninSmallHolderCashews

* Adding plot to BigEarthNet

* Doctstring adjustment for BigEarthNet plot

* Cleaning up ugly test

* Cleaning up bigearthnet classes

* Added time step plot to benin_cashews

* Formatting

* Update benin cashew tests

* Add S1 plotting and type to np.ndarray

* Trying numpy with quotes
* Fix doc link

* Brute forcing rst syntax

* Adding more docs to ChesapeakeCVPR
@github-actions github-actions bot added the testing Continuous integration testing label Dec 29, 2021
torchgeo/datasets/sen12ms.py Show resolved Hide resolved
torchgeo/datasets/sen12ms.py Outdated Show resolved Hide resolved
torchgeo/datasets/sen12ms.py Outdated Show resolved Hide resolved
torchgeo/datasets/sen12ms.py Outdated Show resolved Hide resolved
torchgeo/datasets/sen12ms.py Outdated Show resolved Hide resolved
torchgeo/datasets/sen12ms.py Outdated Show resolved Hide resolved
torchgeo/datasets/sen12ms.py Outdated Show resolved Hide resolved
* Add plotting method for CV4A Kenya Crop Type Dataset

* remove print statements, still fix test_plot

* fix rgb plot test

* fix rgb plot test

* requested changes
@adamjstewart adamjstewart marked this pull request as draft December 29, 2021 17:16
@adamjstewart
Copy link
Collaborator

Can you resolve the merge conflicts?

nilsleh and others added 2 commits December 30, 2021 12:05
* add plotting method

* implement test

* axis off

* prediction flag

* requested changes

* indexing fix
* Refactor RegressionTask testing

* Programmatically determine max value

* Refactor ClassificationTask testing

* Silence warnings

* Refactor SegmentationTask testing

* Fix training mappings

* Fix GeoDataset trainers

* Fix ETCI trainer fake data

* Update OSCD training data

* Get LandCoverAI tests to pass

* Fix OSCD checksum handling

* Fix NAIP-Chesapeake tests

* Fix OSCD tests

* Keep BoundingBox icy

* Fix other datamodules

* Fix chesapeake testing

* Refactor BYOLTask tests

* Style fixes

* Silence pytorch-lightning warnings

* Get coverage for Chesapeake CVPR prior

* Fix trainer tests
@github-actions github-actions bot added datamodules PyTorch Lightning datamodules documentation Improvements or additions to documentation losses Geospatial loss functions models Models and pretrained weights samplers Samplers for indexing datasets trainers PyTorch Lightning trainers transforms Data augmentation transforms labels Dec 30, 2021
@adamjstewart
Copy link
Collaborator

Looks like a bad rebase, this PR now modifies 300+ files.

@nilsleh
Copy link
Collaborator Author

nilsleh commented Dec 30, 2021

Should I delete this branch and open it again with the changes specific to Sen12MS?

@adamjstewart
Copy link
Collaborator

Yeah, if you don't know git well that's probably the easiest approach.

@adamjstewart
Copy link
Collaborator

Closing in favor of #338

@adamjstewart adamjstewart removed this from the 0.2.0 milestone Dec 31, 2021
adamjstewart pushed a commit that referenced this pull request Dec 31, 2021
yichiac pushed a commit to yichiac/torchgeo that referenced this pull request Apr 29, 2023
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 losses Geospatial loss functions models Models and pretrained weights samplers Samplers for indexing datasets testing Continuous integration testing trainers PyTorch Lightning trainers transforms Data augmentation transforms
Projects
None yet
Development

Successfully merging this pull request may close these issues.

7 participants