GID-15

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

Bases: NonGeoDataset

GID-15 dataset.

The GID-15 dataset is a dataset for semantic segmentation.

Dataset features:

• images taken by the Gaofen-2 (GF-2) satellite over 60 cities in China

• masks representing 15 semantic categories

• three spectral bands - RGB

• 150 with 3 m per pixel resolution (6800x7200 px)

Dataset format:

• images are three-channel pngs

• masks are single-channel pngs

• colormapped masks are 3 channel tifs

Dataset classes:

1. background

2. industrial_land

3. urban_residential

4. rural_residential

5. traffic_land

6. paddy_field

7. irrigated_land

8. dry_cropland

9. garden_plot

10. arbor_woodland

11. shrub_land

12. natural_grassland

13. artificial_grassland

14. river

15. lake

16. pond

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

• https://doi.org/10.1016/j.rse.2019.111322

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

Initialize a new GID-15 dataset instance.

Parameters:

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

• split (str) – one of “train”, “val”, 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.

__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]

__len__()

Return the number of data points in the dataset.

Returns:

length of the dataset

Return type:

int

plot(sample, suptitle=None)

Plot a sample from the dataset.

Parameters:

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

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

Returns:

a matplotlib Figure with the rendered sample

Return type:

Figure

Added in version 0.2.