geowatch.tasks.invariants.iarpa_dataset module

class geowatch.tasks.invariants.iarpa_dataset.kwcoco_dataset(coco_dset, sensor='S2', bands='all', patch_size=64, mode='train')[source]

Bases: Dataset

S2_channel_names = ['coastal', 'blue', 'green', 'red', 'B05', 'B06', 'B07', 'nir', 'B09', 'cirrus', 'swir16', 'swir22', 'B8A']
L8_channel_names = ['coastal', 'lwir11', 'lwir12', 'blue', 'green', 'red', 'nir', 'swir16', 'swir22', 'pan', 'cirrus']
common_channel_names = ['coastal', 'blue', 'green', 'red', 'nir', 'swir16', 'swir22', 'cirrus']
get_img(idx, device=None)[source]
num_channels()[source]