Source code for aitlas.datasets.uc_merced

from .multiclass_classification import MultiClassClassificationDataset


LABELS = [
    "agricultural",
    "airplane",
    "baseballdiamond",
    "beach",
    "buildings",
    "chaparral",
    "denseresidential",
    "forest",
    "freeway",
    "golfcourse",
    "harbor",
    "intersection",
    "mediumresidential",
    "mobilehomepark",
    "overpass",
    "parkinglot",
    "river",
    "runway",
    "sparseresidential",
    "storagetanks",
    "tenniscourt",
]


class UcMercedDataset(MultiClassClassificationDataset):

    labels = LABELS
    name = "UC Merced dataset"

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