ADVANCE

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

Bases: NonGeoDataset

ADVANCE dataset.

The ADVANCE dataset is a dataset for audio visual scene recognition.

Dataset features:

• 5,075 pairs of geotagged audio recordings and images

• three spectral bands - RGB (512x512 px)

• 10-second audio recordings

Dataset format:

• images are three-channel jpgs

• audio files are in wav format

Dataset classes:

0. airport

1. beach

2. bridge

3. farmland

4. forest

5. grassland

6. harbour

7. lake

8. orchard

9. residential

10. sparse shrub land

11. sports land

12. train station

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

• https://doi.org/10.1007/978-3-030-58586-0_5

Note

This dataset requires the following additional library to be installed:

• scipy to load the audio files to tensors

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

Initialize a new ADVANCE dataset instance.

Parameters:

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

• 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:

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

• DependencyNotFoundError – If scipy is not installed.

__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

__annotate_func__()

The type of the None singleton.

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 string to use as a suptitle

Returns:

a matplotlib Figure with the rendered sample

Return type:

Figure

Added in version 0.2.