# Autogenerated via: # python ~/code/watch/dev/maintain/mirror_package_geowatch.py def __getattr__(key): import geowatch.tasks.rutgers_material_change_detection.datasets.iarpa_sc_kwdataset as mirror return getattr(mirror, key) def __dir__(): import geowatch.tasks.rutgers_material_change_detection.datasets.iarpa_sc_kwdataset as mirror return dir(mirror) if __name__ == '__main__': # Get root directory of dataset on local machine. from geowatch.tasks.rutgers_material_change_detection.utils.util_paths import get_dataset_root_dir from geowatch.tasks.rutgers_material_change_detection.utils.util_misc import generate_video_slice_object dset_name = "iarpa_drop2v2" root_dir = get_dataset_root_dir(dset_name) # Get dataset split and task_mode. split = "valid" task_mode = "total_sem_change" # Test loading first and last frames of training split. video_slice = generate_video_slice_object(height=300, width=300, n_frames=5, scale=1) dset = IARPA_SC_EVAL_DATASET(root_dir, split, video_slice, task_mode, channels="RGB_NIR", normalize_mode=None) example = dset.__getitem__(0)