PatternNet

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

Bases: NonGeoClassificationDataset

PatternNet dataset.

The PatternNet dataset is a dataset for remote sensing scene classification and image retrieval.

Dataset features:

• 30,400 images with 6-50 cm per pixel resolution (256x256 px)

• three spectral bands - RGB

• 38 scene classes, 800 images per class

Dataset format:

• images are three-channel jpgs

Dataset classes:

0. airplane

1. baseball_field

2. basketball_court

3. beach

4. bridge

5. cemetery

6. chaparral

7. christmas_tree_farm

8. closed_road

9. coastal_mansion

10. crosswalk

11. dense_residential

12. ferry_terminal

13. football_field

14. forest

15. freeway

16. golf_course

17. harbor

18. intersection

19. mobile_home_park

20. nursing_home

21. oil_gas_field

22. oil_well

23. overpass

24. parking_lot

25. parking_space

26. railway

27. river

28. runway

29. runway_marking

30. shipping_yard

31. solar_panel

32. sparse_residential

33. storage_tank

34. swimming_pool

35. tennis_court

36. transformer_station

37. wastewater_treatment_plant

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

• https://doi.org/10.1016/j.isprsjprs.2018.01.004

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

Initialize a new PatternNet 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.

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

Plot a sample from the dataset.

Parameters:

• sample (dict[str, Any]) – a sample returned by NonGeoClassificationDataset.__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

Added in version 0.2.