geowatch.tasks.landcover.model_info module

class geowatch.tasks.landcover.model_info.ModelInfo[source]

Bases: ABC

abstractmethod create_dataset(coco_dset: CocoDataset | str) → Dataset[source]

Create a torch dataset compatible with this model using a CocoDataset :Parameters: coco_dset – CocoDataset of string filepath

Returns: torch dataset

abstract property model_outputs
abstractmethod load_model(weights_filename: Path, device)[source]
class geowatch.tasks.landcover.model_info.S2ModelInfo[source]

Bases: ModelInfo

This model was trained on 13-band Sentinel-2 data with 5 segmentation classes

create_dataset(coco_dset)[source]
property model_outputs
load_model(weights_filename, device)[source]
class geowatch.tasks.landcover.model_info.WVModelInfo[source]

Bases: ModelInfo

This model was trained on 8-band WorldView-3 data with 5 segmentation classes

create_dataset(coco_dset)[source]
property model_outputs
load_model(weights_filename, device)[source]
geowatch.tasks.landcover.model_info.lookup_model_info(weights_filename: Path) → ModelInfo[source]