src.FRAME_FM.dataloaders.demo_geotiff_dataloader

Classes

XarrayStaticDataModule	A simple DataModule for loading tiled static data from a geotiff file using xarray.
GeotiffSpatialDataModule	Module for loading tiled values and their positions from a geotiff file, using xarray.
GeotiffBoundedDataModule	Module for loading tiled values and tile coordinate bounds from a geotiff file, using xarray.

Functions

convert_to_long_lat(x, y, src_crs[, dst_crs])
main()	Try creating the data module and plotting examples. Currently not using hydra.

Module Contents

src.FRAME_FM.dataloaders.demo_geotiff_dataloader.convert_to_long_lat(x, y, src_crs, dst_crs='EPSG:4326')

class src.FRAME_FM.dataloaders.demo_geotiff_dataloader.XarrayStaticDataModule(data_root: str, batch_size: int = 32, num_workers: int = 4, pin_memory: bool = True, persistent_workers: bool = False, train_split: float = 0.85, val_split: float = 0.15, test_split: float = 0.0, split_strategy: str = 'fraction', train_transforms: Callable | None = None, val_transforms: Callable | None = None, test_transforms: Callable | None = None, tile_size: int = 256)

Bases: src.FRAME_FM.utils.LightningDataModuleWrapper.BaseDataModule

A simple DataModule for loading tiled static data from a geotiff file using xarray.

train_dataset: src.FRAME_FM.datasets.InputOnly_Dataset.TransformedInputCoordsDataset

val_dataset: src.FRAME_FM.datasets.InputOnly_Dataset.TransformedInputCoordsDataset

test_dataset: src.FRAME_FM.datasets.InputOnly_Dataset.TransformedInputCoordsDataset | None

tile_size = 256

tile_array(array: xarray.DataArray) → xarray.DataArray

class src.FRAME_FM.dataloaders.demo_geotiff_dataloader.GeotiffSpatialDataModule(data_root: str, batch_size: int = 32, num_workers: int = 4, pin_memory: bool = True, persistent_workers: bool = False, train_split: float = 0.85, val_split: float = 0.15, test_split: float = 0.0, split_strategy: str = 'fraction', train_transforms: Callable | None = None, val_transforms: Callable | None = None, test_transforms: Callable | None = None, tile_size: int = 256)

Bases: XarrayStaticDataModule

Module for loading tiled values and their positions from a geotiff file, using xarray.

extract_position_tensor(array: xarray.DataArray) → torch.Tensor

class src.FRAME_FM.dataloaders.demo_geotiff_dataloader.GeotiffBoundedDataModule(data_root: str, batch_size: int = 32, num_workers: int = 4, pin_memory: bool = True, persistent_workers: bool = False, train_split: float = 0.85, val_split: float = 0.15, test_split: float = 0.0, split_strategy: str = 'fraction', train_transforms: Callable | None = None, val_transforms: Callable | None = None, test_transforms: Callable | None = None, tile_size: int = 256)

Bases: GeotiffSpatialDataModule

Module for loading tiled values and tile coordinate bounds from a geotiff file, using xarray.

extract_position_tensor(tiles: xarray.DataArray) → torch.Tensor

src.FRAME_FM.dataloaders.demo_geotiff_dataloader.main()

Try creating the data module and plotting examples. Currently not using hydra.