micro_sam.training

1"""Functionality for training Segment Anything.
2"""
3
4from .sam_trainer import SamTrainer, SamLogger
5from .util import ConvertToSamInputs, get_trainable_sam_model, identity
6from .joint_sam_trainer import JointSamTrainer, JointSamLogger
7from .simple_sam_trainer import SimpleSamTrainer, MedSAMTrainer
8from .semantic_sam_trainer import SemanticSamTrainer, SemanticMapsSamTrainer
9from .training import train_sam, train_sam_for_configuration, default_sam_loader, default_sam_dataset, CONFIGURATIONS