GeoNRW

class torchgeo.datasets.GeoNRW(root='data', split='train', transforms=None, download=False, checksum=False)

Bases: NonGeoDataset

GeoNRW dataset.

This datasets contains RGB, DEM and segmentation label data from North Rhine-Westphalia, Germany.

Dataset features:

• 7298 training and 485 test samples

• RGB images, 1000x1000px normalized to [0, 1]

• DEM images, unnormalized

• segmentation labels

Dataset format:

• RGB images are three-channel jp2

• DEM images are single-channel tif

• segmentation labels are single-channel tif

Dataset classes:

0. background

1. forest

2. water

3. agricultural

4. residential,commercial,industrial

5. grassland,swamp,shrubbery

6. railway,trainstation

7. highway,squares

8. airport,shipyard

9. roads

10. buildings

Additional information about the dataset can be found on this site.

If you use this dataset in your research, please cite the following paper:

• https://ieeexplore.ieee.org/document/9406194

Added in version 0.6.

__init__(root='data', split='train', transforms=None, download=False, checksum=False)

Initialize the GeoNRW dataset.

Parameters:

• root (str | PathLike[str]) – root directory where dataset can be found

• split (str) – one of “train”, or “test”

• transforms (Callable[[dict[str, Any]], dict[str, Any]] | None) – a function/transform that takes input sample and its target as entry and returns a transformed version

• download (bool) – if True, download dataset and store it in the root directory

• checksum (bool) – if True, check the MD5 of the downloaded files (may be slow)

Raises:

• AssertionError – if split argument is invalid

• DatasetNotFoundError – If dataset is not found and download is False.

__len__()

Return the number of data points in the dataset.

Returns:

length of the dataset

Return type:

int

__getitem__(index)

Return an index within the dataset.

Parameters:

index (int) – index to return

Returns:

data and label at that index

Return type:

dict[str, Any]

plot(sample, show_titles=True, suptitle=None)

Plot a sample from the dataset.

Parameters:

• sample (dict[str, Any]) – a sample returned by __getitem__()

• show_titles (bool) – flag indicating whether to show titles above each panel

• suptitle (str | None) – optional suptitle to use for figure

Returns:

a matplotlib Figure with the rendered sample

Return type:

Figure

__annotate_func__()

The type of the None singleton.