NASA Marine Debris#

class torchgeo.datasets.NASAMarineDebris(root='data', transforms=None, download=False)[source]#

Bases: NonGeoDataset

NASA Marine Debris dataset.

The NASA Marine Debris dataset is a dataset for detection of floating marine debris in satellite imagery.

Dataset features:

  • 707 patches with 3 m per pixel resolution (256x256 px)

  • three spectral bands - RGB

  • 1 object class: marine_debris

  • images taken by Planet Labs PlanetScope satellites

  • imagery taken from 2016-2019 from coasts of Greece, Honduras, and Ghana

Dataset format:

  • images are three-channel geotiffs in uint8 format

  • labels are numpy files (.npy) containing bounding box (xyxy) coordinates

  • additional: images in jpg format and labels in geojson format

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

Note

This dataset requires the following additional library to be installed:

  • azcopy: to download the dataset from Source Cooperative.

Added in version 0.2.

__init__(root='data', transforms=None, download=False)[source]#

Initialize a new NASA Marine Debris 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

Raises:

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

__getitem__(index)[source]#

Return an index within the dataset.

Parameters:

index (int) – index to return

Returns:

data and labels at that index

Return type:

dict[str, Any]

__len__()[source]#

Return the number of data points in the dataset.

Returns:

length of the dataset

Return type:

int

plot(sample, show_titles=True, suptitle=None)[source]#

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