diff --git a/alfie/self_heatmap.py b/alfie/self_heatmap.py index a13ea71..50793bc 100644 --- a/alfie/self_heatmap.py +++ b/alfie/self_heatmap.py @@ -254,7 +254,7 @@ def compute_pixel_diffused_heat_map(self, latent_pixel_id: The latent pixel id from which the heatmap will be diffused throughout the latent image method: Currently only has `thresholding` where at each step to remove the misguiding/noisy pixels to prevent them from focussing/enhancing wrong heatmaps, we use simple thresholding - n_iter: For how many interations to refine the heatmap as described above + n_iter: For how many interactions to refine the heatmap as described above plot_itermediate: If the intermediate heatmaps need to plotted to show the evolutation """ @@ -288,7 +288,7 @@ def compute_diffused_heat_map(self, `n_iter` number of iterations. Returns a SelfGlobalHeatMap collection of the attention diffused heatmaps for all pixels method: Currently only has `thresholding` where at each step to remove the misguiding/noisy pixels to prevent them from focussing/enhancing wrong heatmaps, we use simple thresholding - n_iter: For how many interations to refine the heatmap as described above + n_iter: For how many interactions to refine the heatmap as described above """ # convert the latent 2d image from height.width x height x weight to 1 x height.weight x height x weight @@ -346,4 +346,4 @@ def __iter__(self): def clear(self): self.ids_to_heatmaps.clear() - self.ids_to_num_maps.clear() \ No newline at end of file + self.ids_to_num_maps.clear()