"""SpaceNet2 (GeoBench V2) benchmark dataset."""
from .base import BandSpec
from .geobench_v2 import _V2Dataset
[docs]
class SpaceNet2(_V2Dataset):
"""WorldView building footprint segmentation (3 classes).
8 multispectral + 1 panchromatic band from WorldView satellite.
"""
band_order_strategy = "by_sensor"
name = "spacenet2"
task = "segmentation"
num_classes = 3
multilabel = False
rgb_bands = ["red", "green", "blue"]
split_sizes = {"train": 5186, "val": 1461, "test": 2961}
# fmt: off
bands = [
BandSpec("worldview", "coastal", "coastal", mean=296.081, std=107.482, min=0, max=1227),
BandSpec("worldview", "blue", "blue", mean=357.957, std=151.518, min=0, max=1570),
BandSpec("worldview", "green", "green", mean=465.239, std=229.433, min=0, max=2047),
BandSpec("worldview", "yellow", "yellow", mean=417.796, std=230.014, min=0, max=2047),
BandSpec("worldview", "red", "red", mean=334.455, std=198.499, min=0, max=1933),
BandSpec("worldview", "red_edge", "red_edge", mean=409.533, std=212.211, min=0, max=2047),
BandSpec("worldview", "nir1", "nir1", mean=481.216, std=240.981, min=0, max=2047),
BandSpec("worldview", "nir2", "nir2", mean=364.308, std=196.878, min=0, max=2047),
BandSpec("pan", "pan", "pan", mean=469.092, std=266.975, min=0, max=2047),
]
# fmt: on