microsoft / torchgeo

TorchGeo: datasets, samplers, transforms, and pre-trained models for geospatial data
https://www.osgeo.org/projects/torchgeo/
MIT License
2.66k stars 328 forks source link

Add stitching utilities #30

Open adamjstewart opened 3 years ago

adamjstewart commented 3 years ago

We should add a torchgeo.utils package that provides utilities for common operations, including stitching together patches to create a prediction on an entire tile. This is equivalent to torchvision.utils. See https://arxiv.org/pdf/1805.12219.pdf for a survey of common stitching techniques that we should support. This includes clipping (best default) and averaging. May also want to add weighted averaging. Let's see what libraries exist for this. If rasterio can do this for us for free, that would be awesome.

Z-Zheng commented 3 years ago

See here https://github.com/BloodAxe/pytorch-toolbelt