rai_toolbox.augmentations.fourier.create_heatmaps

rai_toolbox.augmentations.fourier.create_heatmaps(dataloader, image_height_width, *, model, metrics, basis_norm, rand_flip_per_channel, post_pert_batch_transform=None, device=None, row_col_coords=None, factor_2pi_phase_shift=0)