sc2bench.models.segmentation.registry

Functions

get_segmentation_model(cls_or_func_name, ...)	Gets a semantic segmentation model.
load_segmentation_model(model_config, device)	Loads a semantic segmentation model.
register_segmentation_model_class(cls)	Registers a semantic segmentation model class.
register_segmentation_model_func(func)	Registers a function to build a semantic segmentation model.