ggt.visualization.spatial_transform
Module Contents
Functions
|
|
|
Visualize the transformation performed by the spatial transformer |
Attributes
- 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'