Skip to content

TorchGeo Single Tile Embeddings#22

Open
isaaccorley wants to merge 5 commits into
mainfrom
isaac/torchgeo-embeddings
Open

TorchGeo Single Tile Embeddings#22
isaaccorley wants to merge 5 commits into
mainfrom
isaac/torchgeo-embeddings

Conversation

@isaaccorley
Copy link
Copy Markdown
Collaborator

@isaaccorley isaaccorley commented Jul 20, 2025

This PR adds a notebook, notebooks/torchgeo_tile_embeddings.ipynb that introduces a workflow for generating embeddings from Sentinel-2 data using TorchGeo, including:

  • A custom Sentinel2AnnualMosaic dataset class for handling Sentinel-2 annual mosaics.
  • A GridGeoSampler for sampling geospatial tiles and a TorchGeo model (MOSAIKS RCF) for embedding generation.
  • Saving embeddings and geometries as a GeoDataFrame and exporting them to a Parquet file.

@isaaccorley isaaccorley requested a review from cr458 July 20, 2025 01:59
@isaaccorley isaaccorley self-assigned this Jul 20, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant