modlee.model.image_model module

Modlee model for images.

class modlee.model.image_model.ImageCallback(num_classes=2, *args, **kwargs)

Bases: Callback

Saves accuracy

on_validation_batch_end(trainer: Trainer, pl_module: LightningModule, outputs: Tensor | Dict[str, Any] | None, batch: Any, batch_idx: int, dataloader_idx: int = 0) → None

Called when the validation batch ends.

class modlee.model.image_model.ModleeImageModel(task='classification', num_classes=None, *args, **kwargs)

Bases: ModleeModel

Subclass of ModleeModel with image-specific convenience wrappers

• Logs classification accuracy

• Calculates data-specific data statistics

configure_callbacks()

Configure image-specific callbacks.