ggt.visualization.spatial_transform

Module Contents

Functions

visualize_spatial_transform(model, loader, output_dir)

main(model_type, model_path, cutout_size, channels, ...)

Visualize the transformation performed by the spatial transformer

Attributes

log_fmt

ggt.visualization.spatial_transform.visualize_spatial_transform(model, loader, output_dir, device='cpu', nrow=6, return_grids=False, matplotlib_backend='agg')
ggt.visualization.spatial_transform.main(model_type, model_path, cutout_size, channels, n_out, data_dir, split_slug, split, batch_size, nrow, n_workers, normalize)

Visualize the transformation performed by the spatial transformer module.

ggt.visualization.spatial_transform.log_fmt = '%(asctime)s - %(name)s - %(levelname)s - %(message)s'