Skip to content

SegmentationLogger

SegmentationLogger class

dlf.callbacks.segmentation_logger.SegmentationLogger(
    num_classes, num_visualizations=50, opacity=0.5, log_train_images=False
)

Tensorboard visualization for segmentation tasks

Aliases

  • segmentation_logger
  • SegmentationLogger

Arguments

  • num_classes: int. Number of classes, required for confusion matrix
  • num_visualizations: int, optional. Number of images to visualize. Defaults to 50.
  • opacity: float. Opacity of segmentation overlay. Defaults to 0.5.
  • log_train_images: bool. If true training images are visualized as well. Defaults to False.

YAML Configuration

callbacks:
    SegmentationLogger:
        num_classes: 7
        num_visualizations: 200

Example