vis4d.engine.callbacks.visualizer

This module contains utilities for callbacks.

Classes

VisualizerCallback(*args, visualizer[, ...])

Callback for model visualization.

class VisualizerCallback(*args, visualizer, visualize_train=False, show=False, save_to_disk=True, save_prefix=None, output_dir=None, **kwargs)

Callback for model visualization.

__init__(*args, visualizer, visualize_train=False, show=False, save_to_disk=True, save_prefix=None, output_dir=None, **kwargs)

Init callback.

Parameters:

• visualizer (Visualizer) – Visualizer.

• visualize_train (bool) – If the training data should be visualized. Defaults to False.

• show (bool) – If the visualizations should be shown. Defaults to False.

• save_to_disk (bool) – If the visualizations should be saved to disk. Defaults to True.

• save_prefix (str) – Output directory prefix for distinguish different visualizations.

• output_dir (str) – Output directory for saving the visualizations.

setup(trainer, pl_module, stage)

Setup callback.

Return type:

None

on_train_batch_end(trainer, pl_module, outputs, batch, batch_idx)

Hook to run at the end of a training batch.

Return type:

None

on_validation_batch_end(trainer, pl_module, outputs, batch, batch_idx, dataloader_idx=0)

Hook to run at the end of a validation batch.

Return type:

None

on_test_batch_end(trainer, pl_module, outputs, batch, batch_idx, dataloader_idx=0)

Hook to run at the end of a testing batch.

Return type:

None

save(cur_iter, stage)

Save the visualizer state.

Return type:

None