RESISC45

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

Bases: NonGeoClassificationDataset

NWPU-RESISC45 dataset.

The RESISC45 dataset is a dataset for remote sensing image scene classification.

Dataset features:

• 31,500 images with 0.2-30 m per pixel resolution (256x256 px)

• three spectral bands - RGB

• 45 scene classes, 700 images per class

• images extracted from Google Earth from over 100 countries

• images conditions with high variability (resolution, weather, illumination)

Dataset format:

• images are three-channel jpgs

Dataset classes:

0. airplane

1. airport

2. baseball_diamond

3. basketball_court

4. beach

5. bridge

6. chaparral

7. church

8. circular_farmland

9. cloud

10. commercial_area

11. dense_residential

12. desert

13. forest

14. freeway

15. golf_course

16. ground_track_field

17. harbor

18. industrial_area

19. intersection

20. island

21. lake

22. meadow

23. medium_residential

24. mobile_home_park

25. mountain

26. overpass

27. palace

28. parking_lot

29. railway

30. railway_station

31. rectangular_farmland

32. river

33. roundabout

34. runway

35. sea_ice

36. ship

37. snowberg

38. sparse_residential

39. stadium

40. storage_tank

41. tennis_court

42. terrace

43. thermal_power_station

44. wetland

This dataset uses the train/val/test splits defined in the “In-domain representation learning for remote sensing” paper:

• https://arxiv.org/abs/1911.06721

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

• https://doi.org/10.1109/jproc.2017.2675998

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

Initialize a new RESISC45 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:

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

Returns:

a matplotlib Figure with the rendered sample

Return type:

Figure

Added in version 0.2.

__annotate_func__()

The type of the None singleton.