TorchGeo

Core features

• Geospatial Datasets and Samplers

  • 20+ data loaders for satellite imagery and masks
  • Automatic CRS reprojection and resampling
  • Support for multimodal learning and data fusion

• Benchmark Datasets

  • 50+ data loaders for benchmark datasets
  • Quickly experiment with a variety of models
  • Automatically download most datasets

• Pre-Trained Weights

  • 40+ models pre-trained on geospatial data
  • Native support for multispectral imagery
  • Enables transfer learning on smaller datasets

• Reproducibility with Lightning

  • Builtin train/val/test splits and data augmentation
  • Classification, regression, detection, segmentation
  • Command-line interface, support for YAML files