DIOR#
- class torchgeo.datasets.DIOR(root='data', split='train', transforms=None, download=False, checksum=False)[source]#
Bases:
NonGeoDatasetDIOR dataset.
DIOR dataset contains horizontal bounding box annotations of Google Earth Aerial RGB imagery. The test split does not contain bounding box annotations and labels.
Dataset features:
20 classes
192,472 manually annotated bounding box instances
Dataset format:
Images are three channel .jpg files.
Annotations are in Pascal VOC XML format
Classes:
Airplane
Airport
Baseball Field
Basketball Court
Bridge
Chimney
Dam
Expressway Service Area
Expressway Toll Station
Golf Field
Ground Track Field
Harbor
Overpass
Ship
Stadium
Storage Tank
Tennis Court
Train Station
Vehicle
Windmill
If you use this dataset in your research, please cite the following paper:
Added in version 0.7.
- __init__(root='data', split='train', transforms=None, download=False, checksum=False)[source]#
Initialize a new DIOR dataset instance.
- Parameters:
root (str | PathLike[str]) – root directory where dataset can be found
split (Literal['train', 'val', 'test']) – split of the dataset to use, one of ‘train’, ‘val’, ‘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 or corrupted and download is False.
AssertionError – If split argument is invalid.
- __len__()[source]#
Return the number of data points in the dataset.
- Returns:
length of the dataset
- Return type:
- __annotate_func__()#
The type of the None singleton.