torchgeo.samplers¶

Samplers¶

Samplers are used to index a dataset, retrieving a single query at a time. For NonGeoDataset, dataset objects can be indexed with integers, and PyTorch’s builtin samplers are sufficient. For GeoDataset, dataset objects require a bounding box for indexing. For this reason, we define our own GeoSampler implementations below. These can be used like so:

from torchgeo.datasets import Landsat
from torchgeo.samplers import RandomGeoSampler

dataset = Landsat(...)
sampler = RandomGeoSampler(dataset, size=256, length=10000)

This data loader will return 256x256 px images, and has an epoch length of 10,000.

Random Geo Sampler¶

class torchgeo.samplers.RandomGeoSampler(dataset, size, length, roi=None, units=Units.PIXELS)

Bases: GeoSampler

Samples elements from a region of interest randomly.

This is particularly useful during training when you want to maximize the size of the dataset and return as many random chips as possible.

This sampler is not recommended for use with tile-based datasets. Use RandomBatchGeoSampler instead.

__init__(dataset, size, length, roi=None, units=Units.PIXELS)[source]

Initialize a new Sampler instance.

The size argument can either be:

• a single float - in which case the same value is used for the height and width dimension

• a tuple of two floats - in which case, the first float is used for the height dimension, and the second float for the width dimension

Parameters
• dataset (GeoDataset) – dataset to index from

• size (Union[Tuple[float, float], float]) – dimensions of each patch

• length (int) – number of random samples to draw per epoch

• roi (Optional[BoundingBox]) – region of interest to sample from (minx, maxx, miny, maxy, mint, maxt) (defaults to the bounds of dataset.index)

• units (Units) – defines if size is in pixel or CRS units

Changed in version 0.3: Added units parameter, changed default to pixel units

__iter__()[source]

Return the index of a dataset.

Returns

(minx, maxx, miny, maxy, mint, maxt) coordinates to index a dataset

Return type
__len__()[source]

Return the number of samples in a single epoch.

Returns

length of the epoch

Return type

int

Grid Geo Sampler¶

class torchgeo.samplers.GridGeoSampler(dataset, size, stride, roi=None, units=Units.PIXELS)

Bases: GeoSampler

Samples elements in a grid-like fashion.

This is particularly useful during evaluation when you want to make predictions for an entire region of interest. You want to minimize the amount of redundant computation by minimizing overlap between chips.

Usually the stride should be slightly smaller than the chip size such that each chip has some small overlap with surrounding chips. This is used to prevent stitching artifacts when combining each prediction patch. The overlap between each chip (chip_size - stride) should be approximately equal to the receptive field of the CNN.

Note that the stride of the final set of chips in each row/column may be adjusted so that the entire tile is sampled without exceeding the bounds of the dataset.

Let $$i$$ be the size of the input tile. Let $$k$$ be the requested size of the output patch. Let $$s$$ be the requested stride. Let $$o$$ be the number of output rows/columns sampled from each tile. $$o$$ can then be computed as:

$o = \left\lceil \frac{i - k}{s} \right\rceil + 1$

This is almost identical to relationship 5 in https://doi.org/10.48550/arXiv.1603.07285. However, we use ceiling instead of floor because we want to include the final remaining chip.

__init__(dataset, size, stride, roi=None, units=Units.PIXELS)[source]

Initialize a new Sampler instance.

The size and stride arguments can either be:

• a single float - in which case the same value is used for the height and width dimension

• a tuple of two floats - in which case, the first float is used for the height dimension, and the second float for the width dimension

Parameters

Changed in version 0.3: Added units parameter, changed default to pixel units

__iter__()[source]

Return the index of a dataset.

Returns

(minx, maxx, miny, maxy, mint, maxt) coordinates to index a dataset

Return type
__len__()[source]

Return the number of samples over the ROI.

Returns

number of patches that will be sampled

Return type

int

Pre-chipped Geo Sampler¶

class torchgeo.samplers.PreChippedGeoSampler(dataset, roi=None, shuffle=False)

Bases: GeoSampler

Samples entire files at a time.

This is particularly useful for datasets that contain geospatial metadata and subclass GeoDataset but have already been pre-processed into chips.

This sampler should not be used with NonGeoDataset. You may encounter problems when using an ROI that partially intersects with one of the file bounding boxes, when using an IntersectionDataset, or when each file is in a different CRS. These issues can be solved by adding padding.

__init__(dataset, roi=None, shuffle=False)[source]

Initialize a new Sampler instance.

Parameters
• dataset (GeoDataset) – dataset to index from

• roi (Optional[BoundingBox]) – region of interest to sample from (minx, maxx, miny, maxy, mint, maxt) (defaults to the bounds of dataset.index)

• shuffle (bool) – if True, reshuffle data at every epoch

New in version 0.3.

__iter__()[source]

Return the index of a dataset.

Returns

(minx, maxx, miny, maxy, mint, maxt) coordinates to index a dataset

Return type
__len__()[source]

Return the number of samples over the ROI.

Returns

number of patches that will be sampled

Return type

int

Batch Samplers¶

When working with large tile-based datasets, randomly sampling patches from each tile can be extremely time consuming. It’s much more efficient to choose a tile, load it, warp it to the appropriate coordinate reference system (CRS) and resolution, and then sample random patches from that tile to construct a mini-batch of data. For this reason, we define our own BatchGeoSampler implementations below. These can be used like so:

from torchgeo.datasets import Landsat
from torchgeo.samplers import RandomBatchGeoSampler

dataset = Landsat(...)
sampler = RandomBatchGeoSampler(dataset, size=256, batch_size=128, length=10000)

This data loader will return 256x256 px images, and has a batch size of 128 and an epoch length of 10,000.

Random Batch Geo Sampler¶

class torchgeo.samplers.RandomBatchGeoSampler(dataset, size, batch_size, length, roi=None, units=Units.PIXELS)

Bases: BatchGeoSampler

Samples batches of elements from a region of interest randomly.

This is particularly useful during training when you want to maximize the size of the dataset and return as many random chips as possible.

__init__(dataset, size, batch_size, length, roi=None, units=Units.PIXELS)[source]

Initialize a new Sampler instance.

The size argument can either be:

• a single float - in which case the same value is used for the height and width dimension

• a tuple of two floats - in which case, the first float is used for the height dimension, and the second float for the width dimension

Parameters
• dataset (GeoDataset) – dataset to index from

• size (Union[Tuple[float, float], float]) – dimensions of each patch

• batch_size (int) – number of samples per batch

• length (int) – number of samples per epoch

• roi (Optional[BoundingBox]) – region of interest to sample from (minx, maxx, miny, maxy, mint, maxt) (defaults to the bounds of dataset.index)

• units (Units) – defines if size is in pixel or CRS units

Changed in version 0.3: Added units parameter, changed default to pixel units

__iter__()[source]

Return the indices of a dataset.

Returns

batch of (minx, maxx, miny, maxy, mint, maxt) coordinates to index a dataset

Return type
__len__()[source]

Return the number of batches in a single epoch.

Returns

number of batches in an epoch

Return type

int

Base Classes¶

If you want to write your own custom sampler, you can extend one of these abstract base classes.

Geo Sampler¶

class torchgeo.samplers.GeoSampler(dataset, roi=None)

Bases: Sampler[BoundingBox], ABC

Abstract base class for sampling from GeoDataset.

Unlike PyTorch’s Sampler, GeoSampler returns enough geospatial information to uniquely index any GeoDataset. This includes things like latitude, longitude, height, width, projection, coordinate system, and time.

__init__(dataset, roi=None)[source]

Initialize a new Sampler instance.

Parameters
• dataset (GeoDataset) – dataset to index from

• roi (Optional[BoundingBox]) – region of interest to sample from (minx, maxx, miny, maxy, mint, maxt) (defaults to the bounds of dataset.index)

abstract __iter__()[source]

Return the index of a dataset.

Returns

(minx, maxx, miny, maxy, mint, maxt) coordinates to index a dataset

Return type

Batch Geo Sampler¶

class torchgeo.samplers.BatchGeoSampler(dataset, roi=None)

Bases: Sampler[List[BoundingBox]], ABC

Abstract base class for sampling from GeoDataset.

Unlike PyTorch’s BatchSampler, BatchGeoSampler returns enough geospatial information to uniquely index any GeoDataset. This includes things like latitude, longitude, height, width, projection, coordinate system, and time.

__init__(dataset, roi=None)[source]

Initialize a new Sampler instance.

Parameters
• dataset (GeoDataset) – dataset to index from

• roi (Optional[BoundingBox]) – region of interest to sample from (minx, maxx, miny, maxy, mint, maxt) (defaults to the bounds of dataset.index)

abstract __iter__()[source]

Return a batch of indices of a dataset.

Returns

batch of (minx, maxx, miny, maxy, mint, maxt) coordinates to index a dataset

Return type

Units¶

By default, the size parameter specifies the size of the image in pixel units. If you would instead like to specify the size in CRS units, you can change the units parameter like so:

from torchgeo.datasets import Landsat
from torchgeo.samplers import RandomGeoSampler, Units

dataset = Landsat(...)
sampler = RandomGeoSampler(dataset, size=256 * 30, length=10000, units=Units.CRS)