Source code for aitlas.datasets.optimal_31

from .multiclass_classification import MultiClassClassificationDataset


LABELS = [
    "airplane",
    "airport",
    "baseball_diamond",
    "basketball_court",
    "beach",
    "bridge",
    "chaparral",
    "church",
    "circular_farmland",
    "commercial_area",
    "dense_residential",
    "desert",
    "forest",
    "freeway",
    "golf_course",
    "ground_track_field",
    "harbor",
    "industrial_area",
    "intersection",
    "island",
    "lake",
    "meadow",
    "medium_residential",
    "mobile_home_park",
    "mountain",
    "overpass",
    "parking_lot",
    "railway",
    "rectangular_farmland",
    "roundabout",
    "runway",
]


class Optimal31Dataset(MultiClassClassificationDataset):

    url = "https://drive.google.com/file/d/1Fk9a0DW8UyyQsR8dP2Qdakmr69NVBhq9/view"

    labels = LABELS
    name = "Optimal31 dataset"

    def __init__(self, config):
        # now call the constructor to validate the schema and load the data
        super().__init__(config)